Skip to content
New issue

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

#

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

Already on GitHub? # to your account

Jump to bottom

无法使用自己训练后的网络 #18

Open
ZhangGanWei opened this issue Apr 13, 2024 · 3 comments
Open

无法使用自己训练后的网络 #18

ZhangGanWei opened this issue Apr 13, 2024 · 3 comments

Comments

@ZhangGanWei
Copy link

自己训练的权重导出onnx后，运行程序报错，用yolov5s.onnx程序正常，请问怎么使用自己训练的网络？
1712972844023
@ZhangGanWei
Copy link
Author

ZhangGanWei commented Apr 13, 2024 •
edited
Loading

@hpc203 训练时使用的网络结构是yolov5-6.1的原结构，感谢回复

@ZhangGanWei
Copy link
Author

补充：只有用作者的onnx文件才能readNet，原yolov5s权重转化为onnx也报同样错误。

@ZhangGanWei
Copy link
Author

image
image

可视化模型后发现与作者的不一样，按照作者对export.py和yolo.py Detect的修改后也失败了

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

No branches or pull requests
1 participant
@ZhangGanWei