打开微信,使用扫一扫进入页面后,点击右上角菜单,
点击“发送给朋友”或“分享到朋友圈”完成分享
你好,我正在尝试使用pytorch docker进行yolov5的推理和模型转换。
我的运行环境为:win10,docker v4.11.1 ubuntu18.04的官方镜像。没有寒武纪的卡,我在代码中添加了
ct.set_device(-1) 这行代码
/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结束后打印的日志。
还劳烦帮我看下这个问题呢。
热门帖子
精华帖子