search

VeriSilicon / acuity-models

Acuity Model Zoo
https://verisilicon.github.io/acuity-models/
134 stars 23 forks source link

YoloV5 #7

Open fangichao opened 3 years ago

fangichao commented 3 years ago

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?

geneyoung commented 2 years ago

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.