今天出一期在windows、linux ubantu系统中部署yolo or rtdetr项目的流程,供小白快速入门!(声明:以下流程均在蓝耘GPU平台和魔鬼面具的rtdetr项目下进行,其他项目如yolo系列均可使用该方法) 一、
今天出一期在windows、linux & ubantu系统中部署yolo or rtdetr项目的流程,供小白快速入门!(声明:以下流程均在蓝耘GPU平台和魔鬼面具的rtdetr项目下进行,其他项目如yolo系列均可使用该方法) 一、准备工作在此之前需要提前下载以下工具 1.显卡资源(自用电脑 或 服务器平台 2.Pycharm专业版,Vscode,或GPU平台的Jupyter界面(以下教程在Pycharm专业版中展示) 3.XFTP文件传输助手(个人习惯,可使用其他文件传输软件,较为方便) 相关链接附文章末尾 二、pycharm连接远程服务器方法以下流程均在b站有对应视频 ssh -p 20733 root@link.lanyun.net 1.打开pycharm专业版,依次点击工具---部署--配置 2.添加新配置,按以下图片流程添加ssh信息,测试连接(不懂请看b站视频) 3.打开远程主机Remote Host(远程服务器目录) 4.打开远程终端 三、项目环境配置---RTDETR 和YOLO环境相同1.本地需要提前准备好Conda环境以便安装虚拟环境及torch及后续操作
2.云GPU服务器直接选定以下torch和cuda版本直接部署,或创建虚拟环境后按装下方相关包。以下代码装好虚拟环境后可直接全部复制进去,或者一条一条复制,一般不会出现报错情况。
3.装好环境的同时将魔导的项目文件上传进去,可以用XFTP上传或直接上传首先使用XFTP连接远程服务器 四、训练教程---RTDETR 和YOLO相同1.终端下解压项目文件 ②cd RTDETR-main/ (cd到文件目录下) ③修改数据集路径(使用项目默认数据集测试,修改后上传保存) ④运行python train.py 指令提示报错,原因是因为路径错误,datasets出现了两次 ⑤vim /root/.config/Ultralytics/settings.yaml (---vim到上方报错路径) ⑥再次运行python train.py 指令后成功运行,或者可能会出现字体下载卡顿
最后成功运行代码 后续更换数据集的话,则按照以上格式,自行根据实际情况更改数据集即可 五、训练过程中的一些小技巧1.使用nohup指令进行后台训练,防止因断网导致的中断等情况
2.查看是否跑完 nvidia-smi,//查看显存变化,如果跑一个实验占了8个G显存,然后显存变成0的时候就说明跑完了 3.查看跑到哪一轮了,精度是多少 查看runs/train/exp文件里面的result.csv,主要查看map50 和map50-95 的涨点情况 4.查看代码是否报错,打开日志log.log查看代码报错情况 5.运行一次代码,会出现一个进程号,若代码开始跑了,想提前中断的话,请使用ps aux 指令查看PID进程号了,并且使用kill -9 进程号 中断该进程并释放显存 附上相关链接 1.显卡资源(自用电脑 或 服务器平台)https://cloud.lanyun.net/#/activity?uuid=efa15431a91abaaf0bdeb212fffe685a 2.Pycharm专业版,Vscode,或GPU平台的Jupyter界面(以下教程在Pycharm专业版中展示) 3.XFTP文件传输助手(个人习惯,可使用其他文件传输软件,较为方便) 家庭/学校免费 - NetSarang Website |
2022-11-25
2024-04-30
2024-10-18
2022-08-26
2022-09-23