yolov5n.pt 模型转ncnn后,预测结果为空

[superbayes]

trafficstars

detail | 详细描述 | 詳細な説明

我最近使用yolov5r6官方提供的yolov5n.pt,转换为ncnn模型后,无法预测出物体

我的步骤是: 1: python export.py --weights models/yolov5n.pt --include onnx --train 2: python -m onnxsim models/yolov5n.onnx models/yolov5n-sim.onnx 3: 在 转换模型网址 将模型转换为 yolov5n-sim-opt-fp16 模型 4,使用yolov5进行预测,但没有结果 请up有空帮忙看看,万分感谢 yolov5n.zip

[superbayes]

自问自答,解决了,见上图

[cmdbug]

因为代码输入图片是动态尺寸，所以是-1，如果是固定640x640的可以不用改。

[superbayes]

因为代码输入图片是动态尺寸，所以是-1，如果是固定640x640的可以不用改。

@cmdbug 大佬,就是根据你的方案解决了预测不正确的问题 但是,当我用官方提供的模型,训练我自己的数据后,结果又不正常了,debug中....

[Digital2Slave]

自问自答,解决了,见上图

@superbayes 你是用Python调用的么？

[superbayes]

自问自答,解决了,见上图

@superbayes 你是用Python调用的么？

c++调用

[Digital2Slave]

自问自答,解决了,见上图

@superbayes 你是用Python调用的么？

c++调用

嗯嗯，我准备用Python接口，目前没有现成的。我参考yolov5.py自己修改了一个针对YoloV5l的。#4097