Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open
z13228604287 opened this issue Mar 24, 2023 · 3 comments
Open

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

z13228604287 opened this issue Mar 24, 2023 · 3 comments

Comments

@z13228604287
Copy link

python export.py --weights best.pt --include onnx engine --img 640 --device 0
@z13228604287
Copy link
Author

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

@taifyang
Copy link
Owner

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

@monoloxo
Copy link

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@monoloxo @taifyang @z13228604287