×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

pytorch yolov5在docker环境中推理结果不正确 已完结 Jerry30622022-08-23 11:18:38 回复 7 查看 技术答疑
pytorch yolov5在docker环境中推理结果不正确
分享到:

你好,我正在尝试使用pytorch docker进行yolov5的推理和模型转换。

我的运行环境为:win10,docker v4.11.1 ubuntu18.04的官方镜像。没有寒武纪的卡,我在代码中添加了

ct.set_device(-1) 这行代码

在运行https://developer.cambricon.com/index/curriculum/expdetails/id/10/classid/8.html该链接的例子的时候,我按照步骤能够进行模型转换和推理,但是推理结果不正确。例子中能检测到两个人,但是我运行后没有检测出任何东西。

/mapping/Pytorch_Yolov5_Inference/quantize_online/yolo.py:54: TracerWarning: Converting a tensor to a Python integer might cause the trace to be incorrect. We can't record the data flow of Python values, so this value will be treated as a constant in the future. This means that the trace might not generalize to other inputs!

  print('y.shape: ',y.shape)

y.shape:  torch.Size([1, 255, 80, 80])

y.shape:  torch.Size([1, 255, 40, 40])

y.shape:  torch.Size([1, 255, 20, 20])

batchNum: 1

torch.Size([1, 3, 640, 640])

num_boxes_final:  0.4470588266849518

[tensor([], size=(0, 6))]


这是我运行detect.py结束后打印的日志。

还劳烦帮我看下这个问题呢。


版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭