小白复现yolov11训练自己数据集过程中出现自制数据集yaml文件不可用、cuda版本不匹配、vscode实验室服务器ssh不连接以及训练后预测框过大等问题

最新推荐文章于 2025-02-08 01:34:08 发布

五五匠人

最新推荐文章于 2025-02-08 01:34:08 发布

阅读量1k

点赞数 23

文章标签： YOLO python 深度学习服务器 vscode

本文链接：https://blog.csdn.net/weixin_44828282/article/details/143883155

版权

本人参照YOLOv11来了，使用YOLOv11训练自己的数据集和推理(附YOLOv11网络结构图)-CSDN博客对yolov11进行复现自己制作的rgb数据集，复现过程重点出现了自制数据集yaml文件不可用、cuda版本不匹配、vscode实验室服务器ssh不连接以及训练后预测框过大等4个问题，本文对上述问题进行记录，并介绍一下问题形式、可能原因及解决办法，希望能对大家有帮助。

一、自制数据集yaml文件不可用问题

（本阶段训练指令代码是我在自己电脑上试运行，尚未使用实验室服务器）

本人数据集使用labelme进行标注工作，如图

根据网上提供思路，我对yolov11文件夹中的./ultralytics/cfg/datasets/coco.yaml根据自己数据集类别和路径进行更改，并将自制数据集和我更改的配置文件vmd_coco.yaml都放在了coco.yaml配置文件中要求的路径./ultralytics/datasets，如下图部分

运行训练指令（如下）

yolo task=detect mode=train data='D:\Postgraduate Study\Lab Work\Projects\yolov11\ultralytics\datasets\vmd_coco.yaml' epochs=5 batch=64 device=cpu single_cls=False pretrained='.\yolo11s.pt'

报错，提示修改settings.json,以上为错误现象。