利用autodl服务器跑模型

沉靥2024-03-21 13:59

1. 租用服务器

本地改模型
服务器
- 将改进好的、数据集处理好的模型压缩为zip文件
- 上传到阿里云盘
- 打开服务器AUTODL服务器，在主页中选择容器实例
在此位置进行开关机操作，若停止服务器，必须关机，不然会一直扣钱

2. 运行模型

选择AutoPanel
然后选择公共网盘 ，之后选择授权阿里云盘
选择上传好的文件，点击下载
然后选择进入JupyterLab
选择终端
在命令行中使用unzip指令对下载好的压缩文件进行解压（unzip yolov5）
根据解压位置，选择根目录路径，使用cd进入想要的目录层
- cd autodl-tmp
- cd yolov5(文件名)
最后执行代码训练指令（博主自用，可以根据自己文件实际需要对应修改）
- python train.py --data data/zhouzhou128.yaml --cfg models/yolov5s.yaml --weights weights/yolov5s.pt --batch-size 32 --epochs 100
- python train.py --data data/fish.yaml --cfg models/yolov5l-Cneb.yaml --weights weights/yolov5l.pt --batch-size 16 --epochs 150
- 注意这个命令是直接指定的参数，如果想用train里填好的参数，可以直接运行。输入：python train.py
- 此外，注意此命令的路径，如果本地当初各个文件不是放在models或weights等文件夹里的，要重新复制一个过去哦
补充
- 如果遇到/.../.../.../1.jpg类似错误（AssertionError: Image Not Found），删除数据集中labels目录下的cache文件
- 运行模型命令：python 文件名称
- 不能删除文件夹，但可以剪切到别的地方去
- 如果运行train.PY出现这种错误说明镜像环境中的protobuf版本不对可以输入指令 pip install protobuf==3.11.2 重新安装即可（原先的protobuf 或者其他安装包的版本可以通过输入pip list来查看）出现这个报错的时候系统环境中的protobuf版本为4.23.2