利用autodl服务器跑模型

avatar
作者
筋斗云
阅读量:1

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在这里插入图片描述

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!