yolo-world使用自己数据集训练

YOLO-World下载:

https://github.com/AILab-CVC/YOLO-World/tree/master
1.数据准备

数据格式COCO格式即可
2.配置文件修改

configs/finetune_coco/yolo_world_v2_l_vlpan_bn_sgd_1e-3_40e_8gpus_finetune_coco.py
(1)

模型下载路径:

yoloworld:https://hf-mirror.com/wondervictor/YOLO-World

clip-vit-base-patch32:https://hf-mirror.com/openai/clip-vit-base-patch32
(2)

(3)

(4)

3.训练模型

python tools/train.py configs/finetune_coco/yolo_world_v2_l_vlpan_bn_sgd_1e-3_40e_8gpus_finetune_coco.py --work-dir log --amp --resume

相关推荐
EdisonZhou20 小时前
MAF快速入门(4)多Agent工作流编排
llm·aigc·agent·.net core
智泊AI1 天前
AI大模型+数字人 | AI数字人直播带货竟然这么简单!
llm
大模型教程1 天前
爆肝6周,手把手教你搭建一套生产级RAG论文研究助手
程序员·llm·agent
大模型教程1 天前
技术干货丨AI 大模型微调到底是什么?一篇通俗文帮你弄明白
程序员·llm·agent
AI大模型1 天前
别再瞎学大模型了,这份GitHub神级课程火爆全网
程序员·llm·agent
AI大模型1 天前
检索增强生成(RAG)与大语言模型微调(Fine-tuning)的差异、优势及使用场景详解
程序员·llm·agent
Coding茶水间1 天前
基于深度学习的面部口罩检测系统演示与介绍(YOLOv12/v11/v8/v5模型+Pyqt5界面+训练代码+数据集)
图像处理·人工智能·深度学习·yolo·目标检测·计算机视觉
musk12121 天前
YOLOv8n模型微调全指南:从环境搭建到技能储备 (内容由 AI 生成)
人工智能·yolo
用户576905308011 天前
微调入门尝试:沐雪角色扮演
python·llm