yolov5目标检测

yolov5教程

参考：
跟着“我是土堆”UP主的配置
第一步：PyTorch环境的配置及安装
 第二步：目标检测 YOLOv5 开源代码项目调试与讲解实战

PyCharm安装步骤

PyCharm社区版下载地址

安装好之后下载Anaconda，并根据上面的PyTorch配置视频进行环境配置，并在PyCharm中选择虚拟环境

启动Anaconda，切换虚拟环境

如果没有添加Anaconda环境变量需要添加以下几个

常见问题：

RuntimeError: The size of tensor a (80) must match the size of tensor b (56) at non-singleton dimension 3

一般都是5.0版本的，然后把6.0的SPPF代码粘贴到5.0替换SPP之后就会出现此问题，建议直接更换6.0版本，问题直接解决

Traceback (most recent call last):
  File "F:/YOLO-V5/yolov5-5.0/detect.py", line 179, in <module>
    detect()
  File "F:/YOLO-V5/yolov5-5.0/detect.py", line 35, in detect
    model = attempt_load(weights, map_location=device)  # load FP32 model
  File "F:\YOLO-V5\yolov5-5.0\models\experimental.py", line 118, in attempt_load
    ckpt = torch.load(w, map_location=map_location)  # load
  File "E:\python\ancharn1\lib\site-packages\torch\serialization.py", line 607, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "E:\python\ancharn1\lib\site-packages\torch\serialization.py", line 882, in _load
    result = unpickler.load()
  File "E:\python\ancharn1\lib\site-packages\torch\serialization.py", line 875, in find_class
    return super().find_class(mod_name, name)
AttributeError: Can't get attribute 'SPPF' on <module 'models.common' from 'F:\\YOLO-V5\\yolov5-5.0\\models\\common.py'>

将6.0“models\common.py”中的SPPF类替换到5.0的SPP类，一般会在5.0出现这种报错。

PyCharm无法启动Conda虚拟环境解决方法

心和Faerie:
说一下我遇到的坑（WIN10，显卡3060）：

CUDA版本，30系的显卡要选择CUDA的版本高于11
目前yolo有了V6，按教程跑V5会报各种各样的错（如没有SPPF类这种），可以按教程直接跑V6
pycocotools可以直接用pip install
pytorch因为网络原因下载失败或者有多个版本（因为1搞错了），可以尝试手动删除pytorch文件再下载
最后感谢up

yolov5教程

PyCharm安装步骤

常见问题：

演示