【环境配置】YOLOV5训练VisDrone数据集

前言

YOLOv5目前自带VisDrone配置，所以几乎不需要进行更改，clone下来开箱即用即可。
注意：建议目录，新建一个yolov5的文件夹，在此文件夹下面clone yolov5，执行train.py文件，会自动创建datasets文件夹，用来存放VisDrone数据集。

➜ yolov5
├── datasets
└── yolov5-master

这样做的话可以一劳永逸，会自动下载VisDrone数据集并解压。如果你想更改VisDrone数据集的存放位置，可以更改data/VisDrone.yaml文件。

训练命令

git clone https://github.com.cnpmjs.org/ultralytics/yolov5.git
cd yolov5-master
pip install -r requirements.txt
python train.py --data data/VisDrone.yaml --cfg models/yolov5x.yaml --weights weights/yolov5x.pt --device 0,1,2,3

推理命令

python detect.py --weights runs/exp1/weights/best.pt --source inference/images/ --device 0 --save-txt