训练
准备
在AutoDL
中创建实例,我这里的配置:
进入JupyterLab
。
下载yolov5
:https://github.com/ultralytics/yolov5
上传到服务器。
在终端执行:
cd yolov5
unzip yolov5-master.zip
cd yolov5-master
conda init
然后重开终端
cd yolov5-master
pip install -r requirements.txt
上传yolov5s.pt
到yolov5-master
文件夹下,然后执行:
python detect.py # 测试一下能否检测
上传数据集到yolov5-master
文件夹下,然后执行:
unzip datasets.zip
训练
data下的coco128.yaml拷贝一份,重命名为shanyi.yaml
,然后将之前的classes_with_index.txt
的内容放到names
的下面。
path: ./datasets
train: images/train
val: images/val
names:
0: xx
1: xx
train.py
中的data
的yaml
文件夹名改成shanyi.yaml
。
然后执行:
python train.py
等待他慢慢训练。
训练完以后可以将结果下载下来放到本地
tar -cvf runs.tar.gz runs/