Open z13228604287 opened 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
你可以看看你的yolov5代码版本是不是7.0,或者用netron工具比较一下onnx有什么不同
这个因该是cpu版本吧,--device 0就是导出的gpu了吧
python export.py --weights best.pt --include onnx engine --img 640 --device 0