zackxue

onnx export can work

1

python ./models/export.py --weights ./weights/helmet_head_person_s.pt --img 640 --batch 1 File "demo_onnx.py", line 301, in detections = detect_onnx(official=False, image_path=image_path) File "demo_onnx.py", line 180, in detect_onnx session = onnxruntime.InferenceSession('./weights/helmet_head_person_s.onnx') File "C:\Users\vision\Anaconda3\envs\pt_gpu\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 283,...