onnx/safety_detect.py
文件名从 safety_detect.py 修改 @@ -18,7 +18,7 @@ return self.detect_objects(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()
