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
7条答案
按热度按时间6tr1vspr1#
自问自答,解决了,见上图
2ic8powd2#
因为代码输入图片是动态尺寸，所以是-1，如果是固定640x640的可以不用改。
vbopmzt13#
因为代码输入图片是动态尺寸，所以是-1，如果是固定640x640的可以不用改。
@cmdbug 大佬,就是根据你的方案解决了预测不正确的问题
但是,当我用官方提供的模型,训练我自己的数据后,结果又不正常了,debug中....
os8fio9y4#
自问自答,解决了,见上图
@superbayes 你是用Python调用的么？
tyky79it5#
自问自答,解决了,见上图
@superbayes 你是用Python调用的么？
c++调用
4nkexdtk6#
自问自答,解决了,见上图
@superbayes 你是用Python调用的么？
c++调用
嗯嗯，我准备用Python接口，目前没有现成的。我参考 yolov5.py 自己修改了一个针对YoloV5l的。 #4097
s2j5cfk07#
自问自答,解决了,见上图
请问你的5n是哪个版本的？ 6.0版本吗