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

目录

一  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_