[에러코드/DeepSort] KeyError: "The name 'net/images:0' refers to a Tensor which does not exist. The operation, 'net/images', does not exist in the graph."

DeepSort 논문 코드 구현 중 발생한 에러이다.

 

Yolo 또는 다른 Detector를 통해 Dataset에 해당하는 Detection들을 생성할 때 생기는 문제로

 

입력노드와 출력노드의 이름을 변경하여 에러를 해결할 수 있다.

 

tools폴더 안에 있는 generate_detections.py의 83번째 줄과 85번째 줄을 다음과 같이 수정하자.

 

"net / % s : 0"=> "% s : 0"

 

class ImageEncoder(object):

    def __init__(self, checkpoint_filename, input_name="images",
                 output_name="features"):
        self.session = tf.compat.v1.Session()
        with tf.io.gfile.GFile(checkpoint_filename, "rb") as file_handle:
            graph_def = tf.compat.v1.GraphDef()
            graph_def.ParseFromString(file_handle.read())
        tf.import_graph_def(graph_def, name="net")
        self.input_var = tf.compat.v1.get_default_graph().get_tensor_by_name(
            "%s:0" % input_name)
        self.output_var = tf.compat.v1.get_default_graph().get_tensor_by_name(
            "%s:0" % output_name)