yolo 入门 2 - traeCN 帮助你进行模型训练

yolo 还真不怕你把照片拍的很大,它有办法自动处理这些事情,但是我们在采集图片时,还是尽量确保所有的图片大小差不多,这样也方便打标签。

打标签工具:labelimg (工具和数据集相见网盘:链接: https://pan.baidu.com/s/1QbPyoGq3_yOpWYvorxyPeg?pwd=8vmg 提取码: 8vmg)

也可以使用pip install labelimg 安装,但windows 用户建议还是使用现成的 labelimg.exe,实测发现 pip 安装的易出现闪退,非常影响效率和心情。

(python特产,包兼容性问题)

要根治,最好是下载python 源码,安装源码指定的依赖版本

labeimg 操作详见:https://zhuanlan.zhihu.com/p/550021453

打标签前,建议把所有的图片放在一个目录下(对,可以混放),但是要避免文件名冲突。

解决第一个问题,文件名冲突。

问题场景,我有两个目录,这两个目录里面都是图片,但是图片的文件名有部分冲突,我想放到一起,怎么办。

(来无中生有)

(图片生成完毕)

总结:把你想做的事情,尽量描述清楚,剩下的交给 traeCN

接下来,你可以开始打标签了。

打好标签后,在原有图片目录基础上,你会得到一个标签目录,标签目录内的文件名,和图片目录完全一致。

接下来:

"我有一个图片目录 imgClass,一个标签目录 labels,我要按照 yolov8 的目录格式将图片集重新组织,如图所示。你给我写个代码,将图片输出到 target 目录,还要产生一个符合 yolov8 的 data.yaml文件。训练集、验证集、测试集的分割比例为 8:1:1,分割前,你要打乱图片的顺序,确保图片足够分散。"

traeCN 的操作如下图(使用微信、QQ 等工具截好图,直接粘贴就行,它支持图片内容识别):

很快搞定

接下来微调一下即可:

真实的 data.yaml 长这样:

如果你有多个分类,要按上面的格式,注意标签txt 文件里面,0 对应的是第一个标签

这个"0" 实际上是 'five'

(yolov8 格式数据集示例详见网盘,可以直接使用:链接: https://pan.baidu.com/s/1QbPyoGq3_yOpWYvorxyPeg?pwd=8vmg 提取码: 8vmg)

最后,训练模型。

训练开始前,确认你已经配置好了环境。

AI学习环境配置合集

https://blog.csdn.net/razelan/article/details/152720013?spm=1001.2014.3001.5502

注意:如果你有 n 卡,这里要改一下:

注意:在 python 3.10.x 版本 yolo 最多安装到 yolov11(对应ultralytics 8.3.0)。如果你用的是英伟达 50 系列显卡,要安装 torch2.8.0 cuda 以上版本。

在此之前,记得: pip 更换清华源

最后:出 bug 了怎么办?

先选中出错内容:

点击"添加到对话"

traeCN 很强大,它可以解决的问题不限于 traeCN 以内,甚至可以解决一些系统上的问题,灵活运用能让你事半功倍。

甚至,你可以给它提问,让它回答,而不是写代码。

最后,这玩意免费的。

相关推荐
羊羊小栈6 天前
基于「YOLO目标检测 + 多模态AI分析」的水稻病害智能检测分析预警系统(LangChain框架)
人工智能·yolo·目标检测·毕业设计·创业创新·大作业
YOLO_DATA7 天前
遥感滑坡检测的数据集 2299 张 1类 yolo格式 遥感滑坡检测数据集
人工智能·深度学习·yolo·计算机视觉·无人机·yolo数据集·ai数据集
计算机编程-吉哥7 天前
YOLO26 vs YOLO11 vs YOLOv8:深度学习咖啡果实成熟度分割系统【计算机毕业设计选题推荐】
人工智能·python·深度学习·yolo·django·毕业设计
西西弗Sisyphus7 天前
YOLO26 图像分割 - 训练的数据加载:取图、在线增强与标签缓存
yolo·yolo26
YOLO_DATA8 天前
无人机高速公路道路缺陷数据集 道路损伤数据集 公路裂缝识别 AI大疆数据集 10798期
人工智能·深度学习·yolo·机器学习·cnn
YOLO数据集集合8 天前
小目标无人机、飞机、直升机检测数据集 | 小目标检测 无人机检测 空中目标识别 低空安防 反无人机9090期
深度学习·yolo·目标检测·无人机·飞机·小目标·直升机
YOLO数据集集合9 天前
全球无人机航拍树冠覆盖数据集 | 树冠分割 语义分割 无人机航拍 全球森林监测 GeoTIFF 生态恢复9083期
深度学习·yolo·目标检测·无人机·无人机视角·全球森林
Python图像识别9 天前
10-【2027毕设】YOLO11PCB缺陷检测识别系统 - Python完整源码+PyQt5界面+训练模型+数据集
python·深度学习·yolo·毕业设计·毕设
YOLO数据集集合9 天前
无人机树木识别检测数据集 | 树木检测 树种分类 无人机航拍 林业调查9082期
深度学习·yolo·目标检测·无人机·无人机视角·树冠识别·树冠
qq7422349849 天前
YOLO 之外的另一半:RF-DETR 检测实战
yolo