YOLO V7 detect.py运行后图像没Bounding Box问题解决方案

本来以为电脑有啥问题,查了一下可能用half这个GPU加速导致转完数据都变成NAN了。原因参考如下:

pytorch半精度half计算loss得到nan的可能解决方法_York1996的博客-CSDN博客

原问题

YOLO V7 detect.py运行后图像没Bounding Box问题解决方案_第1张图片

1.可以使用cpu去跑  用下面这句。

python detect.py --weights yolov7.pt --conf 0.25 --img-size 640 --device cpu --source inference/images/bus.jpg

  或者在detect.py中这句改一下  half 改成False

YOLO V7 detect.py运行后图像没Bounding Box问题解决方案_第2张图片

2. 改一下torch版本  

torch 1.12.1
torchaudio 0.12.1
torchvision 0.13.1

ps:我还没试

不行  我试了没用!!

3.换一个更大内存的显卡……

等我换了试试

你可能感兴趣的:(python,yolov7,python,深度学习,tensorflow)