如何用xshell连接矩池云,并用矩池云训练自己的模型

235 阅读1分钟

xshell和xftp文件 链接:pan.baidu.com/s/1tEDry0zo… 提取码:esda --来自百度网盘超级会员V4的分享

1,完成上一步我说的标注功能

如何使用labelimg进行数据标注,并放入yolo进行训练 - 掘金 (juejin.cn)

2,获得矩池云的GPU的host和密码

image.png

3,连接成功之后,把自己本地的yolov5文件丢到矩池云中

image.png

image.png

4,进入到云端GPU的yolov5文件夹,并安装环境依赖

pip install -i  https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt 

image.png

5,安装好之后启动train.py文件进行模型训练

image.png

python train.py

image.png 已经在训练了!

6,训练好之后,模型会保存在如图所示的文件夹中

image.png

7,把模型从云端下载下来

image.png

8,在本机跑detect.py 文件,进行推理

exp6/weights/best.pt 就是我们训练好的模型

9,本地也装下环境,这里不再赘述

10,本地运行detect.py进行推理

image.png 根据图片来改 效果在这里

image.png