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

python opencv加载onnx

Open yezhugongjue opened this issue 2 years ago • 6 comments

程序跑通了 调试代码也可以看到具体检测 person tie 但是最后出图 没有框,想请教下是哪里出了问题

yezhugongjue avatar Aug 10 '22 14:08 yezhugongjue

person:0.88 person:0.88 person:0.87 person:0.80 tie:0.69 tie:0.69 tie:0.68 person:0.64 person:0.64 person:0.63 person:0.37 person:0.36 我用 官方的方式导出python export.py --weights yolov5s.pt --include torchscript onnx 映入模型打了下检测,发现他重复检测了好多次 但是去网盘下载 yolov5s.onnx就ok了,请教大牛啥原因

yezhugongjue avatar Aug 11 '22 09:08 yezhugongjue

person:0.88 person:0.88 person:0.87 person:0.80 tie:0.69 tie:0.69 tie:0.68 person:0.64 person:0.64 person:0.63 person:0.37 person:0.36 我用 官方的方式导出python export.py --weights yolov5s.pt --include torchscript onnx 映入模型打了下检测,发现他重复检测了好多次 但是去网盘下载 yolov5s.onnx就ok了,请教大牛啥原因

你用 官方的方式导出python export.py --weights yolov5s.pt --include torchscript onnx 里面包含了yolo层的,而我的仓库里的代码是默认onnx文件里不包含yolo层的,把代码里的yolo层decode去掉就可以的

hpc203 avatar Sep 15 '22 06:09 hpc203

person:0.88 person:0.88 person:0.87 person:0.80 tie:0.69 tie:0.69 tie:0.68 person:0.64 person:0.64 person:0.63 person:0.37 person:0.36 我用 官方的方式导出python export.py --weights yolov5s.pt --include torchscript onnx 映入模型打了下检测,发现他重复检测了好多次 但是去网盘下载 yolov5s.onnx就ok了,请教大牛啥原因

你用 官方的方式导出python export.py --weights yolov5s.pt --include torchscript onnx 里面包含了yolo层的,而我的仓库里的代码是默认onnx文件里不包含yolo层的,把代码里的yolo层decode去掉就可以的

请问是在哪里将yolo层的decode去掉啊,参考了知乎上的代码后,发现那个部分train变量被删去了

ygyg99 avatar Jul 13 '23 05:07 ygyg99

同问请问老哥解决了吗?

HOJULIN avatar Aug 09 '23 07:08 HOJULIN

同问请问老哥解决了吗?

解决了?是那个部分的yolo的decode啊?

chengdashia avatar Nov 01 '23 05:11 chengdashia

请问有解决吗 @chengdashia @ygyg99 @HOJULIN

ZhangGanWei avatar Apr 13 '24 07:04 ZhangGanWei