yolo v5.50 坑

最新推荐文章于 2024-07-20 08:31:38 发布

diyer_zhou

最新推荐文章于 2024-07-20 08:31:38 发布

阅读量943

点赞数 1

分类专栏： YOLO 文章标签： python pycharm

本文链接：https://blog.csdn.net/diyer_zhou/article/details/120615878

版权

YOLO 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

博主在国庆期间成功安装并运行了YOLO模型，但遇到GPU运行时无法识别图片和视频的问题，尽管摄像头识别正常。经过排查，发现是代码中针对YouTube视频处理的部分存在问题，缺少对URL是否存在的判断。解决方案是添加URL检查，并导入和使用pafy库获取视频最佳格式的URL。

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

国庆装了5天 YOLO终于能跑了。过程没记录。

现在还存在GPU跑的时候图片和视频没法识别的问题，摄像头显示是可以识别的。

Model Summary: 476 layers, 87730285 parameters, 0 gradients, 218.8 GFLOPS
Traceback (most recent call last):
File "C:\Users\diyer_zhou\PycharmProjects\yolov5-5.0\detect.py", line 179, in <module>
detect()
File "C:\Users\diyer_zhou\PycharmProjects\yolov5-5.0\detect.py", line 51, in detect
dataset = LoadStreams(source, img_size=imgsz, stride=stride)
File "C:\Users\diyer_zhou\PycharmProjects\yolov5-5.0\utils\datasets.py", line 279, in __init__
if 'youtube.com/' in url or 'youtu.be/' in url: # if source is YouTube video
TypeError: argument of type 'int' is not iterable
1/1: 0...
Process finished with exit code 1

解决：

增加一个 if url:

 if url:
                if 'youtube.com/' in url or 'youtu.be/' in url:  # if source is YouTube video
                    check_requirements(('pafy', 'youtube_dl'))
                    import pafy
                    url = pafy.new(url).getbest(preftype="mp4").url