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

相关推荐
智驱力人工智能9 小时前
小区高空抛物AI实时预警方案 筑牢社区头顶安全的实践 高空抛物检测 高空抛物监控安装教程 高空抛物误报率优化方案 高空抛物监控案例分享
人工智能·深度学习·opencv·算法·安全·yolo·边缘计算
工程师老罗10 小时前
YOLOv1 核心结构解析
yolo
Lun3866buzha10 小时前
YOLOv10-BiFPN融合:危险物体检测与识别的革新方案,从模型架构到实战部署全解析
yolo
uncle_ll11 小时前
RAG 系统性能跃迁:LlamaIndex 索引优化实战指南
llm·rag·检索·llamaindex
Katecat9966311 小时前
YOLOv8-MambaOut在电子元器件缺陷检测中的应用与实践_1
yolo
工程师老罗12 小时前
YOLOv1 核心知识点笔记
笔记·yolo
无名修道院16 小时前
AI大模型微调-LLM、Token、生成与推理详解
llm·token·ai大模型应用开发
工程师老罗17 小时前
基于Pytorch的YOLOv1 的网络结构代码
人工智能·pytorch·yolo
学习3人组20 小时前
YOLO模型集成到Label Studio的MODEL服务
yolo
孤狼warrior20 小时前
YOLO目标检测 一千字解析yolo最初的摸样 模型下载,数据集构建及模型训练代码
人工智能·python·深度学习·算法·yolo·目标检测·目标跟踪