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

相关推荐
机器之心2 小时前
ICML 2024高分论文 | 零阶优化器微调大模型,大幅降低内存
llm
demo007x2 小时前
🔥如何在LlamaIndex中使用RAG?
人工智能·程序员·llm
大象机器人3 小时前
交互式乘法 myCobot 让学习充满乐趣!
人工智能·python·学习·yolo·机械臂·具身智能
Carota不是菜4 小时前
提示工程总结
人工智能·llm
will-wil7 小时前
理解Attention,MHA、MQA、GQA理论知识和代码实现
人工智能·pytorch·深度学习·llm
Channing Lewis15 小时前
LLM的成功改变了人类理解世界的方式
llm
营赢盈英15 小时前
使用 Ollama 时遇到的问题
python·ai·llm·llama·llama-index
阿_旭19 小时前
基于YOLOv10深度学习的CT扫描图像肾结石智能检测系统【python源码+Pyqt5界面+数据集+训练代码】深度学习实战、目标检测
人工智能·python·深度学习·yolo·yolov10
FL16238631291 天前
[数据集][目标检测]桥梁检测数据集VOC+YOLO格式1116张1类别
深度学习·yolo·目标检测
大耳朵爱学习2 天前
大模型技术:能否引领十年的行业变革与职业发展黄金期?......
人工智能·程序人生·语言模型·自然语言处理·大模型·llm·llama