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

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐