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

王多头发

已于 2022-10-21 13:19:27 修改

阅读量1.7k

点赞数 3

分类专栏： python yolov7 文章标签： python 深度学习 tensorflow

于 2022-10-17 16:53:05 首次发布

本文链接：https://blog.csdn.net/QAQterrible/article/details/127367203

版权

4 篇文章 0 订阅

订阅专栏

1 篇文章 0 订阅

订阅专栏

这篇博客探讨了使用PyTorch进行半精度（half）计算时遇到loss为NAN的问题。可能的解决方法包括切换到CPU运行、回退到特定版本的torch库，或者升级到更高内存的GPU。作者已经尝试了前两种方案但未成功，第三种方案还在测试中。

摘要由CSDN通过智能技术生成

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

原问题

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

2. ~~改一下torch版本~~

torch 1.12.1
torchaudio 0.12.1
torchvision 0.13.1

~~ps：我还没试~~

不行我试了没用！！

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

等我换了试试

关注

专栏目录