【食物识别】训练

训练

准备

在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/