xshell和xftp文件 链接:pan.baidu.com/s/1tEDry0zo… 提取码:esda --来自百度网盘超级会员V4的分享
1,完成上一步我说的标注功能
如何使用labelimg进行数据标注,并放入yolo进行训练 - 掘金 (juejin.cn)
2,获得矩池云的GPU的host和密码
3,连接成功之后,把自己本地的yolov5文件丢到矩池云中
4,进入到云端GPU的yolov5文件夹,并安装环境依赖
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt
5,安装好之后启动train.py文件进行模型训练
python train.py
已经在训练了!
6,训练好之后,模型会保存在如图所示的文件夹中
7,把模型从云端下载下来
8,在本机跑detect.py 文件,进行推理
exp6/weights/best.pt 就是我们训练好的模型
9,本地也装下环境,这里不再赘述
10,本地运行detect.py进行推理
根据图片来改
效果在这里