前言:租服务器要比在本地配置环境再去训练yolo模型简单的多,希望大家可以耐心看下去。

一、所用到的软件安装

1.vscode安装

(1)去b站官网上随便找个安装视频看就行,非常简单,就不说了

(2)安装好之后,打开vscode,安装远程所需的插件

插件安装好之后,会发现左侧多了个电脑的图标。如下图所示。

2.FileZilla安装

安装链接: https://pan.baidu.com/s/1PYDXUKYxlgpNFbGiB31OYQ?pwd=kwtz 提取码: kwtz

3.yolov8下载

下载yolov8链接: https://pan.baidu.com/s/1i7QbKjG26KFe-BPf5KvZdQ?pwd=p5sh 提取码: p5sh

二、具体步骤

1.打开AutoDL官网,注册账号并且充值

根据自己需求充值,10元,20元等等。

2.租用服务器并且选择环境

咱们现在跑这个yolov8的代码,组一个便宜点的就行,基本上都能跑起来,地区可以随便选,主要看你想租什么样性能的GPU。

根据上图选择镜像,然后创建并开机就行。之后就会跳转到包含登录指令和密码的界面,如下图所示

3.打开vscode,连接远程服务器

首先复制autodl官网上的登陆指令,打开vscode,点击左侧电脑图标,点击加号,在输入框中粘贴登陆指令。

 然后选择Liunx系统并输入密码,需要重复输入数次,运气好可能就输入一次即可,输入两次,三次大家也不用担心。只要最后连接成功即可。连接成功如下图所示。

4.传输数据到服务器

打开FileZilla软件, 点击左上角打开站点管理器建立一个新站点。

 打开vscode,打开SSH配置文件,找到主机名和端口号。

在FileZilla软件选择ssh协议并将ip地址、端口号、用户名、密码等填入其中(这部分与配置文件中一样),然后点击连接,点击保存密码,点击总是信任该主机,点击确定连接上服务器。

连接成功后,在root目录下新建一个code的文件夹。

将yolov8.zip拖到这个code文件夹中,此时,已经将yolov8压缩包上传到服务器。

5.在服务器中解压缩,并且打开文件夹

在vscode中新建终端

在终端输入以下命令,进入到yolov8.zip所在的目录下

cd /root/code

解压缩

unzip yolov8.zip

打开yolov8文件夹,可能也得需要输入密码,输入即可。

6.使用pip安装YOLOv8

在终端运行下述命令

pip install ultralytics

7.开始训练

yolo task=detect mode=train model=yolov8n.pt data=dataset/data.yaml epochs=20 imgsz=640 batch=8 device=0

训练成功,接下来检验训练成果。

8.查看模型推理结果

训练完成之后,喂模型图片,看是否能准确识别。在yolov8文件中,我已经在文件夹test_images中加入测试图片,所以现在只需运行

yolo task=detect mode=predict model=runs/detect/train13/weights/best.pt source=test_images/ show=True

运行完成之后,在runs\detect\predict文件夹下查看推理结果。

测试图片都被标注,成功运行。对了,服务器用完记得关机哈,要不会持续扣费。


总结

要是觉得好不错的话,麻烦点个赞,要是能关注一下,那就更好了。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐