YOLOv5-Lite极简使用

初试

clone下来，插上USB摄像头，运行：

python detect.py --source 0

结果报错：

FileNotFoundError: [Errno 2] No such file or directory: 'weights/v5lite-s.pt'

说好的自动下载呢？手动下载了一下，接下来，报错：

File "/YOLOv5-Lite/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

直接去代码把这段注释掉了，反正也不用youtu.be。继续报错，应该是torch的版本问题：

  File "/Users/XXX/anaconda3/envs/yolov5/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1207, in __getattr__
    raise AttributeError("'{}' object has no attribute '{}'".format(
AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'

参考了YOLOv5的一个issue，直接注释掉一个参数：

image.png

终于可以了。

试试树莓派

YOLOv5-Lite主打轻小快，可以部署到树莓派。但是我入手了Jetson了，所以就不试了哈。