yolov5-v6.1-opencv-onnxrun icon indicating copy to clipboard operation
yolov5-v6.1-opencv-onnxrun copied to clipboard

Published 20 hours ago • verified publisher icon hpc203

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

Open ZhangGanWei opened this issue 10 months ago • 3 comments

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

ZhangGanWei avatar Apr 13 '24 01:04 ZhangGanWei

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

ZhangGanWei avatar Apr 13 '24 03:04 ZhangGanWei

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

ZhangGanWei avatar Apr 13 '24 07:04 ZhangGanWei

image image

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

ZhangGanWei avatar Apr 13 '24 07:04 ZhangGanWei