onnx/identifier.py
文件名从 identifier.py 修改 @@ -14,7 +14,7 @@ return self.idengify(image) def initialize_model(self, path): self.session = onnxruntime.InferenceSession(path, providers=['CUDAExecutionProvider', 'CPUExecutionProvider']) self.session = onnxruntime.InferenceSession(path, providers=['CPUExecutionProvider']) self.class_names = eval(self.session.get_modelmeta().custom_metadata_map['names']) # Get model info self.get_input_details()
