【yolov5】开源代码复现（pytorch）目标检测入门

阿宙宙宙

已于 2023-10-13 21:36:04 修改

阅读量676

点赞数 3

分类专栏：目标检测文章标签： YOLO pytorch 人工智能目标检测

于 2023-10-08 12:53:47 首次发布

本文链接：https://blog.csdn.net/qq_39972370/article/details/133677224

版权

目标检测专栏收录该内容

3 篇文章 0 订阅

订阅专栏

环境配置

用yolov5进行预测

训练yolov5模型

环境配置

注意：本文使用yolov5的5.0版本进行演示

github上搜索yolov，选择5.0版本进行下载

解压后用pycharm打开，设置pytorch解释器

pycharm提示缺文件，点击安装

或者在终端里安装：pip install -r requirements.txt

（1）如果作者提供requirments.txt 文件∶

可以利用PyCharm自带的智能提示进行安装

或者利用pip install -r requirements.txt 指令进行安装

（2）如果作者没有提供requirments.txt文件：

根据运行报错信息，百度，手动安装缺少的库

提示安装pycocotools报错的话，可以看下面这篇文章

Failed to build pycocotoolsERROR: Could not build wheels for pycocotools which use PEP 517 and cann_zhangfenger的博客-CSDN博客

将两个文件放到文件夹Anaconda3\envs\pytorch\Lib\site-packages下面

拷贝完后重新执行安装命令：pip install pycocotools -i https://pypi.tuna.tsinghua.edu.cn/simple

即可安装成功。

用yolov5进行预测

在使用代码前可以看看readme文档的介绍

运行detect.py文件，里面有main函数

下载文件报错，直接点进链接下载（慢的话，可以复制到迅雷下载）

运行还是报错，因为我的yolov是5.0版本，上面链接下载的yolov5s.pt不是5.0版本的，所以不匹配

重新用下面这个链接下载

https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt

运行后还是报错

出现报错AttributeError: Cant get attribute SPPF on module models

解决方案

找到model/common.py文件

发现没有SPPF这个类，把下面代码直接复制到common.py里面即可，记得把import warnings放在最上面

import warnings
class SPPF(nn.Module):
    def __init__(self, c1, c2, k=5):
        super().__init__()
        c_ = c1 // 2
        self.cv1 = Conv(c1, c_, 1, 1)
        self.cv2 = Conv(c_ * 4, c2, 1, 1)
        self.m = nn.MaxPool2d(kernel_size=k, stride=1, padding=k // 2)
 
    def forward(self, x):
        x = self.cv1(x)
        with warnings.catch_warnings():
            warnings.simplefilter('ignore')
            y1 = self.m(x)
            y2 = self.m(y1)
            return self.cv2(torch.cat([x, y1, y2, self.m(y2)], 1))

再次运行detect.py文件，不再报错