【存档记录】AutoDL云服务器训练yolov5coco128数据

镜像选择：PyTorch 1.7.0
Python 3.8(ubuntu18.04)
Cuda 11.0
开机后：终端命令行输入

环境配置部分

source activate
conda create -n yolov5coco python=3.8
conda activate yolov5coco
pip install torch1.9.0+cu111 torchvision0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

代码及数据集处理部分

1. 下载代码并安装相关文件

git clone https://github.com/ultralytics/yolov5
cd yolov5
pip install -r requirements.txt

2. 下载数据集coco128.zip解压到yolov5目录下
   数据集来源http://t.csdn.cn/sCqgm
   tips：一个小坑coco128.yaml文件需要修改

path: ../datasets/coco128  # dataset root dir

autodl默认…/是从root开始算，这样导致找不到已下载的数据集，改为

path: ../yolov5/datasets/coco128  # dataset root dir

开始训练

python train.py
完成