看了网上好多YOLOv8训练自己数据集的帖子,都存在一个问题,在远程服务器上该怎样部署?下面是我的步骤:
GitHub - ultralytics/ultralytics: NEW - YOLOv8 in PyTorch > ONNX > CoreML > TFLite
新建dataset文件夹,文件夹文件如图所示
其中VOCdevkit中目录,Annotations中存放所有xml标签,JPEGImages中存放所有图像,txt文件中为所有txt标签
images和labels中的文件夹都一样,images中存放图像,labels中存放标签(txt文件)
到此,数据集制作就完成了,如果有想要划分数据集图像的代码,可以关注留言,我可以看到
在yolov8下新建data.yaml
其中的内容如图所示
train: ./dataset/images/train # train images (relative to 'path') 128 images
val: ./dataset/images/val # val images (relative to 'path') 128 images
test: ./dataset/images/test
nc: 3
# Classes
names: ['类别1','类别2','类别3']
自己替换类别和个数
Releases · ultralytics/assets · GitHub
自己下载权重哈!!!一般就是yolov8n.pt, yolov8s.pt , yolov8l.pt, yolov8x.pt,自己需要哪个下载哪个
下载好后,直接放在yolov8下。
具体怎么调,这是个玄学!!!自己把握
在yolov8下新建train.py,内容如下:
from ultralytics import YOLO
model = YOLO("weights/yolov8n.pt")
results = model.train(data="data.yaml",imgsz=640, epochs=300, batch=16, device=0, workers=4)
这里我就用yolov8n了,也就是模型最小,速度最快的那位!!
其次,这里的imgsz根据自己数据集选择,不要太大!!batch一般就是4,16,32,64,自己选。但是越大模型收敛越快!!!works我选择的4,如果自己设备好,可以考虑8个线程。
此处需要将datasets数据集复制到与yolov8同级目录下!!!
原因是遇到以下报错!!!!真的是,恶心至极,竟然找不到数据集!!!原因是因为Ubuntu下的远程服务器中的setting设置导致的,因此,要把数据集dataset复制一份,挪到datasets(自己新建一个)中。
RuntimeError: Dataset 'data.yaml' error ❌
Dataset 'data.yaml' images not found ⚠️, missing paths ['/hy-tmp/datasets/dataset/images/val']
Note dataset download directory is '/hy-tmp/datasets'. You can update this in '/root/.config/Ultralytics/settings.yaml'
可以看到终于运行成功了!!!
对于我们这种科研狗,还是得有这一步的,毕竟这一步可比结果重要
还是在yolov8目录下新建val.py。此时切记,训练的最好的best.pt放在yolov8目录下。这里的batch=1这就是评估每一张图像(切记切记!!),devices=0代表gpu个数
from ultralytics import YOLO
model = YOLO("best.pt") #用于迁移训练的权重文件路径
results = model.val(data="data.yaml",batch=1, device=0,save=True)
最终python val.py运行成功!本人的实验数据保密!!
当然,大家有问题可以在评论区留言,当然有时候可能系统信息太多可能会漏,大家可以在b站发消息给我!如果有必要我会在b站出对应视频教程,需要的可以在下面关注小编,评论区写上视频