search

taifyang / yolov5-cls-det-seg-opencv

15 stars 7 forks source link

为撒 我训练的 转换 onnx 无法读取 #1

Open z13228604287 opened 1 year ago

z13228604287 commented 1 year ago

python export.py --weights best.pt --include onnx engine --img 640 --device 0

z13228604287 commented 1 year ago

PyTorch Hub: model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5s-seg.onnx') # WARNING SegmentationModel not yet supported for PyTorch Hub AutoShape inference

taifyang commented 1 year ago

你可以看看你的yolov5代码版本是不是7.0，或者用netron工具比较一下onnx有什么不同

monoloxo commented 1 year ago

这个因该是cpu版本吧，--device 0就是导出的gpu了吧