1.创建环境 conda create -n yolo python = 3.8
2.安装pytorch1.7以上,这边安装了1.7.1版本的
pip install torch1.7.1+cu101 torchvision0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
用下列指令安装其他包
git clone https://github.com/ultralytics/yolov5
pip install -U -r yolov5/requirements.txt
建议一个一个安装,这样保证不会出错。
3.下载权重,运行检测代码
去官网https://github.com/ultralytics/yolov5/releases
下载自己需要的模型权重,放到./weights里面
然后运行下列代码测试
python detect.py --source data/images --weights yolov5s.pt --conf 0.25
运行成功的话,生成图片放在./runs/detect/exp里面
4.train自己的数据集。
(弄成VOC格式)
首先,自己的训练集格式要有voc的一样
images里面放自己训练的图片
labels里面放yolo格式的txt文件。
下面附xml格式转化为txt格式代码。
impor