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