YoloV5

Open fangichao opened this issue 4 years ago • 1 comments

when i import yolov5 model in step 1, it raise pytorch verison error , yolov5 model trained by torch=1.9, but the tool's requrements.txt requires torch==1.2.0。 can i update torch?

Sep 17 '21 09:09 fangichao

Please refer to https://github.com/ultralytics/yolov5/releases/ to convert pytorch model to onnx model, and then use "pegasus import onnx ..." to import this newly generated onnx model.

Apr 24 '22 10:04 geneyoung