YOLO V5实现及运行问题AttributeError: ‘Detect‘ object has no attribute ‘m‘

最新推荐文章于 2024-05-12 11:18:11 发布

潘旭阳

最新推荐文章于 2024-05-12 11:18:11 发布

阅读量1.6w

点赞数 16

分类专栏：目标检测文章标签： pytorch 深度学习

本文链接：https://blog.csdn.net/Joseph__Lagrange/article/details/108255992

版权

目标检测专栏收录该内容

1 篇文章 1 订阅

订阅专栏

一 YOLO V5 实现

二运行问题AttributeError: 'Detect' object has no attribute 'm'

一 YOLO V5 实现

分为四个部分：环境搭建、源码拉载、权重下载、测试demo

# 1 搭建环境

conda create -n pytorch_1_6   # 创建

conda activate pytorch_1_6  # 激活

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch  # 安装torch

# 2 拉载源码

git clone https://github.com/ultralytics/yolov5.git  # 拉源码

pip install -U -r requirements.txt  # 安装依赖

# 3 权重下载

下面云盘里有权重文件

#  4 测试demo

python detect.py

# 默认情况下，脚本会去读取inference/images下的所有图片并进行目标检测，带有目标框的结果图片保存

# 在inference/out下

注意：（1）linux和win环境搭建过程一样，但是前提都使用conda；pytorch安装要依据自己的机器情况安装；

（2）在拉载源码时会用到git命令，如果没有git命令，可以安装 pip install git；

（3）权重文件有四个文件s、m、l和x，测试性能和存储大小不同；

（4）yolov5可以测试图片、本地摄像头、本地视频等文件

权重文件云盘地址（附有源码）：链接：https://pan.baidu.com/s/1MqKeS5a_FLUV59814evO6A 提取码：0000

参考资料：https://github.com/ultralytics/yolov5 和 https://xugaoxiang.com/2020/06/17/yolov5/

测试结果：（两张图片）

二运行问题AttributeError: 'Detect' object has no attribute 'm'

这个原因是权重文件有问题，从ultralytics的github上下载的yolov5s.pt文件有27M，而在它的子网页下载的yolov5s.pt文件有14.5M，后面这个文件好使，上面云盘里面就是14.5M的yolov5s.pt文件。将14.5M的yolov5s.pt文件放入到项目的weights文件夹下。