├── .dockerignore ├── .gitattributes ├── .gitignore ├── .gitmodules ├── Dockerfile ├── LICENSE ├── README.md ├── assets ├── finetune_yoloworld.png ├── image_prompt.png ├── reparameterize.png ├── yolo_arch.png └── yolo_logo.png ├── configs ├── finetune_coco │ ├── README.md │ ├── yolo_world_l_dual_vlpan_2e-4_80e_8gpus_finetune_coco.py │ ├── yolo_world_l_dual_vlpan_2e-4_80e_8gpus_mask-refine_finetune_coco.py │ ├── yolo_world_l_efficient_neck_2e-4_80e_8gpus_mask-refine_finetune_coco.py │ ├── yolo_world_v2_l_efficient_neck_2e-4_80e_8gpus_mask-refine_finetune_coco.py │ ├── yolo_world_v2_l_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py │ ├── yolo_world_v2_l_vlpan_bn_sgd_1e-3_40e_8gpus_finetune_coco.py │ ├── yolo_world_v2_l_vlpan_bn_sgd_1e-3_80e_8gpus_mask-refine_finetune_coco.py │ ├── yolo_world_v2_m_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py │ ├── yolo_world_v2_s_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py │ ├── yolo_world_v2_s_rep_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py │ ├── yolo_world_v2_s_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py │ ├── yolo_world_v2_x_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py │ └── yolo_world_v2_xl_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py ├── image_prompts │ └── yolo_world_v2_l_vlpan_bn_2e-4_80e_8gpus_image_prompt_demo.py ├── pretrain │ ├── yolo_world_v2_l_clip_large_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_800ft_lvis_minival.py │ ├── yolo_world_v2_l_clip_large_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py │ ├── yolo_world_v2_l_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_1280ft_lvis_minival.py │ ├── yolo_world_v2_l_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py │ ├── yolo_world_v2_l_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_val.py │ ├── yolo_world_v2_m_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_1280ft_lvis_minival.py │ ├── yolo_world_v2_m_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py │ ├── yolo_world_v2_m_vlpan_bn_noeinsum_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py │ ├── yolo_world_v2_s_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_1280ft_lvis_minival.py │ ├── yolo_world_v2_s_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py │ ├── yolo_world_v2_x_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_cc3mlite_train_lvis_minival.py │ ├── yolo_world_v2_x_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_1280ft_lvis_minival.py │ ├── yolo_world_v2_x_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py │ └── yolo_world_v2_xl_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py ├── pretrain_v1 │ ├── README.md │ ├── yolo_world_l_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py │ ├── yolo_world_l_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_val.py │ ├── yolo_world_m_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py │ ├── yolo_world_s_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py │ └── yolo_world_x_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py ├── prompt_tuning_coco │ ├── READEME.md │ ├── yolo_world_v2_l_vlpan_bn_2e-4_80e_8gpus_mask-refine_prompt_tuning_coco.py │ ├── yolo_world_v2_l_vlpan_bn_2e-4_80e_8gpus_prompt_tuning_coco.py │ └── yolo_world_v2_l_vlpan_bn_sgd_1e-3_80e_8gpus_all_finetuning_coco.py └── segmentation │ ├── README.md │ ├── yolo_world_seg_l_dual_vlpan_2e-4_80e_8gpus_allmodules_finetune_lvis.py │ ├── yolo_world_seg_l_dual_vlpan_2e-4_80e_8gpus_seghead_finetune_lvis.py │ ├── yolo_world_seg_m_dual_vlpan_2e-4_80e_8gpus_allmodules_finetune_lvis.py │ ├── yolo_world_seg_m_dual_vlpan_2e-4_80e_8gpus_seghead_finetune_lvis.py │ ├── yolo_world_v2_seg_l_vlpan_bn_2e-4_80e_8gpus_seghead_finetune_lvis.py │ └── yolo_world_v2_seg_m_vlpan_bn_2e-4_80e_8gpus_seghead_finetune_lvis.py ├── data └── texts │ ├── coco_class_texts.json │ ├── lvis_v1_base_class_captions.json │ ├── lvis_v1_class_texts.json │ └── obj365v1_class_texts.json ├── demo ├── README.md ├── gradio_demo.py ├── image_demo.py ├── image_prompt_demo.py ├── inference.ipynb ├── sample_images │ ├── bus.jpg │ └── zidane.jpg ├── simple_demo.py └── video_demo.py ├── deploy ├── __init__.py ├── easydeploy │ ├── README.md │ ├── README_zh-CN.md │ ├── backbone │ │ ├── __init__.py │ │ ├── common.py │ │ └── focus.py │ ├── bbox_code │ │ ├── __init__.py │ │ └── bbox_coder.py │ ├── deepstream │ │ ├── CMakeLists.txt │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── coco_labels.txt │ │ ├── configs │ │ │ ├── config_infer_rtmdet.txt │ │ │ ├── config_infer_yolov5.txt │ │ │ └── config_infer_yolov8.txt │ │ ├── custom_mmyolo_bbox_parser │ │ │ └── nvdsparsebbox_mmyolo.cpp │ │ └── deepstream_app_config.txt │ ├── docs │ │ └── model_convert.md │ ├── examples │ │ ├── config.py │ │ ├── cv2_nms.py │ │ ├── main_onnxruntime.py │ │ ├── numpy_coder.py │ │ ├── preprocess.py │ │ └── requirements.txt │ ├── model │ │ ├── __init__.py │ │ ├── backend.py │ │ ├── backendwrapper.py │ │ └── model.py │ ├── nms │ │ ├── __init__.py │ │ ├── ort_nms.py │ │ └── trt_nms.py │ ├── onnx_demo.py │ └── tools │ │ ├── build_engine.py │ │ ├── export_onnx.py │ │ └── image-demo.py ├── export_onnx.py ├── onnx_demo.py └── tflite_demo.py ├── docs ├── data.md ├── deploy.md ├── faq.md ├── finetuning.md ├── installation.md ├── prompt_yolo_world.md ├── reparameterize.md ├── tflite_deploy.md ├── update_20250123.md ├── updates.md └── yoloworld_v2_results.md ├── pyproject.toml ├── requirements ├── basic_requirements.txt ├── demo_requirements.txt └── onnx_requirements.txt ├── tools ├── dist_test.sh ├── dist_train.sh ├── generate_image_prompts.py ├── generate_text_prompts.py ├── reparameterize_yoloworld.py ├── test.py └── train.py └── yolo_world ├── __init__.py ├── datasets ├── __init__.py ├── mm_dataset.py ├── transformers │ ├── __init__.py │ ├── mm_mix_img_transforms.py │ └── mm_transforms.py ├── utils.py ├── yolov5_cc3m_grounding.py ├── yolov5_lvis.py ├── yolov5_mixed_grounding.py ├── yolov5_obj365v1.py ├── yolov5_obj365v2.py └── yolov5_v3det.py ├── engine ├── __init__.py └── optimizers │ ├── __init__.py │ └── yolow_v5_optim_constructor.py ├── models ├── __init__.py ├── assigner │ ├── __init__.py │ └── task_aligned_assigner.py ├── backbones │ ├── __init__.py │ └── mm_backbone.py ├── data_preprocessors │ ├── __init__.py │ └── data_preprocessor.py ├── dense_heads │ ├── __init__.py │ ├── yolo_world_head.py │ └── yolo_world_seg_head.py ├── detectors │ ├── __init__.py │ ├── yolo_world.py │ └── yolo_world_image.py ├── layers │ ├── __init__.py │ └── yolo_bricks.py ├── losses │ ├── __init__.py │ └── dynamic_loss.py └── necks │ ├── __init__.py │ └── yolo_world_pafpn.py └── version.py /.dockerignore: -------------------------------------------------------------------------------- 1 | docs 2 | Dockerfile -------------------------------------------------------------------------------- /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/.gitmodules -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/README.md -------------------------------------------------------------------------------- /assets/finetune_yoloworld.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/assets/finetune_yoloworld.png -------------------------------------------------------------------------------- /assets/image_prompt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/assets/image_prompt.png -------------------------------------------------------------------------------- /assets/reparameterize.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/assets/reparameterize.png -------------------------------------------------------------------------------- /assets/yolo_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/assets/yolo_arch.png -------------------------------------------------------------------------------- /assets/yolo_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/assets/yolo_logo.png -------------------------------------------------------------------------------- /configs/finetune_coco/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/finetune_coco/README.md -------------------------------------------------------------------------------- /configs/finetune_coco/yolo_world_l_dual_vlpan_2e-4_80e_8gpus_finetune_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/finetune_coco/yolo_world_l_dual_vlpan_2e-4_80e_8gpus_finetune_coco.py -------------------------------------------------------------------------------- /configs/finetune_coco/yolo_world_l_dual_vlpan_2e-4_80e_8gpus_mask-refine_finetune_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/finetune_coco/yolo_world_l_dual_vlpan_2e-4_80e_8gpus_mask-refine_finetune_coco.py -------------------------------------------------------------------------------- /configs/finetune_coco/yolo_world_l_efficient_neck_2e-4_80e_8gpus_mask-refine_finetune_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/finetune_coco/yolo_world_l_efficient_neck_2e-4_80e_8gpus_mask-refine_finetune_coco.py -------------------------------------------------------------------------------- /configs/finetune_coco/yolo_world_v2_l_efficient_neck_2e-4_80e_8gpus_mask-refine_finetune_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/finetune_coco/yolo_world_v2_l_efficient_neck_2e-4_80e_8gpus_mask-refine_finetune_coco.py -------------------------------------------------------------------------------- /configs/finetune_coco/yolo_world_v2_l_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/finetune_coco/yolo_world_v2_l_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py -------------------------------------------------------------------------------- /configs/finetune_coco/yolo_world_v2_l_vlpan_bn_sgd_1e-3_40e_8gpus_finetune_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/finetune_coco/yolo_world_v2_l_vlpan_bn_sgd_1e-3_40e_8gpus_finetune_coco.py -------------------------------------------------------------------------------- /configs/finetune_coco/yolo_world_v2_l_vlpan_bn_sgd_1e-3_80e_8gpus_mask-refine_finetune_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/finetune_coco/yolo_world_v2_l_vlpan_bn_sgd_1e-3_80e_8gpus_mask-refine_finetune_coco.py -------------------------------------------------------------------------------- /configs/finetune_coco/yolo_world_v2_m_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/finetune_coco/yolo_world_v2_m_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py -------------------------------------------------------------------------------- /configs/finetune_coco/yolo_world_v2_s_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/finetune_coco/yolo_world_v2_s_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py -------------------------------------------------------------------------------- /configs/finetune_coco/yolo_world_v2_s_rep_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/finetune_coco/yolo_world_v2_s_rep_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py -------------------------------------------------------------------------------- /configs/finetune_coco/yolo_world_v2_s_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/finetune_coco/yolo_world_v2_s_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py -------------------------------------------------------------------------------- /configs/finetune_coco/yolo_world_v2_x_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/finetune_coco/yolo_world_v2_x_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py -------------------------------------------------------------------------------- /configs/finetune_coco/yolo_world_v2_xl_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/finetune_coco/yolo_world_v2_xl_vlpan_bn_2e-4_80e_8gpus_mask-refine_finetune_coco.py -------------------------------------------------------------------------------- /configs/image_prompts/yolo_world_v2_l_vlpan_bn_2e-4_80e_8gpus_image_prompt_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/image_prompts/yolo_world_v2_l_vlpan_bn_2e-4_80e_8gpus_image_prompt_demo.py -------------------------------------------------------------------------------- /configs/pretrain/yolo_world_v2_l_clip_large_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_800ft_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain/yolo_world_v2_l_clip_large_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_800ft_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain/yolo_world_v2_l_clip_large_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain/yolo_world_v2_l_clip_large_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain/yolo_world_v2_l_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_1280ft_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain/yolo_world_v2_l_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_1280ft_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain/yolo_world_v2_l_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain/yolo_world_v2_l_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain/yolo_world_v2_l_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain/yolo_world_v2_l_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_val.py -------------------------------------------------------------------------------- /configs/pretrain/yolo_world_v2_m_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_1280ft_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain/yolo_world_v2_m_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_1280ft_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain/yolo_world_v2_m_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain/yolo_world_v2_m_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain/yolo_world_v2_m_vlpan_bn_noeinsum_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain/yolo_world_v2_m_vlpan_bn_noeinsum_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain/yolo_world_v2_s_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_1280ft_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain/yolo_world_v2_s_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_1280ft_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain/yolo_world_v2_s_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain/yolo_world_v2_s_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain/yolo_world_v2_x_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_cc3mlite_train_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain/yolo_world_v2_x_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_cc3mlite_train_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain/yolo_world_v2_x_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_1280ft_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain/yolo_world_v2_x_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_1280ft_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain/yolo_world_v2_x_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain/yolo_world_v2_x_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain/yolo_world_v2_xl_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain/yolo_world_v2_xl_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain_v1/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain_v1/README.md -------------------------------------------------------------------------------- /configs/pretrain_v1/yolo_world_l_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain_v1/yolo_world_l_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain_v1/yolo_world_l_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain_v1/yolo_world_l_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_val.py -------------------------------------------------------------------------------- /configs/pretrain_v1/yolo_world_m_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain_v1/yolo_world_m_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain_v1/yolo_world_s_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain_v1/yolo_world_s_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py -------------------------------------------------------------------------------- /configs/pretrain_v1/yolo_world_x_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/pretrain_v1/yolo_world_x_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py -------------------------------------------------------------------------------- /configs/prompt_tuning_coco/READEME.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/prompt_tuning_coco/READEME.md -------------------------------------------------------------------------------- /configs/prompt_tuning_coco/yolo_world_v2_l_vlpan_bn_2e-4_80e_8gpus_mask-refine_prompt_tuning_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/prompt_tuning_coco/yolo_world_v2_l_vlpan_bn_2e-4_80e_8gpus_mask-refine_prompt_tuning_coco.py -------------------------------------------------------------------------------- /configs/prompt_tuning_coco/yolo_world_v2_l_vlpan_bn_2e-4_80e_8gpus_prompt_tuning_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/prompt_tuning_coco/yolo_world_v2_l_vlpan_bn_2e-4_80e_8gpus_prompt_tuning_coco.py -------------------------------------------------------------------------------- /configs/prompt_tuning_coco/yolo_world_v2_l_vlpan_bn_sgd_1e-3_80e_8gpus_all_finetuning_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/prompt_tuning_coco/yolo_world_v2_l_vlpan_bn_sgd_1e-3_80e_8gpus_all_finetuning_coco.py -------------------------------------------------------------------------------- /configs/segmentation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/segmentation/README.md -------------------------------------------------------------------------------- /configs/segmentation/yolo_world_seg_l_dual_vlpan_2e-4_80e_8gpus_allmodules_finetune_lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/segmentation/yolo_world_seg_l_dual_vlpan_2e-4_80e_8gpus_allmodules_finetune_lvis.py -------------------------------------------------------------------------------- /configs/segmentation/yolo_world_seg_l_dual_vlpan_2e-4_80e_8gpus_seghead_finetune_lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/segmentation/yolo_world_seg_l_dual_vlpan_2e-4_80e_8gpus_seghead_finetune_lvis.py -------------------------------------------------------------------------------- /configs/segmentation/yolo_world_seg_m_dual_vlpan_2e-4_80e_8gpus_allmodules_finetune_lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/segmentation/yolo_world_seg_m_dual_vlpan_2e-4_80e_8gpus_allmodules_finetune_lvis.py -------------------------------------------------------------------------------- /configs/segmentation/yolo_world_seg_m_dual_vlpan_2e-4_80e_8gpus_seghead_finetune_lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/segmentation/yolo_world_seg_m_dual_vlpan_2e-4_80e_8gpus_seghead_finetune_lvis.py -------------------------------------------------------------------------------- /configs/segmentation/yolo_world_v2_seg_l_vlpan_bn_2e-4_80e_8gpus_seghead_finetune_lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/segmentation/yolo_world_v2_seg_l_vlpan_bn_2e-4_80e_8gpus_seghead_finetune_lvis.py -------------------------------------------------------------------------------- /configs/segmentation/yolo_world_v2_seg_m_vlpan_bn_2e-4_80e_8gpus_seghead_finetune_lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/configs/segmentation/yolo_world_v2_seg_m_vlpan_bn_2e-4_80e_8gpus_seghead_finetune_lvis.py -------------------------------------------------------------------------------- /data/texts/coco_class_texts.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/data/texts/coco_class_texts.json -------------------------------------------------------------------------------- /data/texts/lvis_v1_base_class_captions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/data/texts/lvis_v1_base_class_captions.json -------------------------------------------------------------------------------- /data/texts/lvis_v1_class_texts.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/data/texts/lvis_v1_class_texts.json -------------------------------------------------------------------------------- /data/texts/obj365v1_class_texts.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/data/texts/obj365v1_class_texts.json -------------------------------------------------------------------------------- /demo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/demo/README.md -------------------------------------------------------------------------------- /demo/gradio_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/demo/gradio_demo.py -------------------------------------------------------------------------------- /demo/image_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/demo/image_demo.py -------------------------------------------------------------------------------- /demo/image_prompt_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/demo/image_prompt_demo.py -------------------------------------------------------------------------------- /demo/inference.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/demo/inference.ipynb -------------------------------------------------------------------------------- /demo/sample_images/bus.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/demo/sample_images/bus.jpg -------------------------------------------------------------------------------- /demo/sample_images/zidane.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/demo/sample_images/zidane.jpg -------------------------------------------------------------------------------- /demo/simple_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/demo/simple_demo.py -------------------------------------------------------------------------------- /demo/video_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/demo/video_demo.py -------------------------------------------------------------------------------- /deploy/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deploy/easydeploy/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/README.md -------------------------------------------------------------------------------- /deploy/easydeploy/README_zh-CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/README_zh-CN.md -------------------------------------------------------------------------------- /deploy/easydeploy/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/backbone/__init__.py -------------------------------------------------------------------------------- /deploy/easydeploy/backbone/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/backbone/common.py -------------------------------------------------------------------------------- /deploy/easydeploy/backbone/focus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/backbone/focus.py -------------------------------------------------------------------------------- /deploy/easydeploy/bbox_code/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/bbox_code/__init__.py -------------------------------------------------------------------------------- /deploy/easydeploy/bbox_code/bbox_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/bbox_code/bbox_coder.py -------------------------------------------------------------------------------- /deploy/easydeploy/deepstream/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/deepstream/CMakeLists.txt -------------------------------------------------------------------------------- /deploy/easydeploy/deepstream/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/deepstream/README.md -------------------------------------------------------------------------------- /deploy/easydeploy/deepstream/README_zh-CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/deepstream/README_zh-CN.md -------------------------------------------------------------------------------- /deploy/easydeploy/deepstream/coco_labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/deepstream/coco_labels.txt -------------------------------------------------------------------------------- /deploy/easydeploy/deepstream/configs/config_infer_rtmdet.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/deepstream/configs/config_infer_rtmdet.txt -------------------------------------------------------------------------------- /deploy/easydeploy/deepstream/configs/config_infer_yolov5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/deepstream/configs/config_infer_yolov5.txt -------------------------------------------------------------------------------- /deploy/easydeploy/deepstream/configs/config_infer_yolov8.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/deepstream/configs/config_infer_yolov8.txt -------------------------------------------------------------------------------- /deploy/easydeploy/deepstream/custom_mmyolo_bbox_parser/nvdsparsebbox_mmyolo.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/deepstream/custom_mmyolo_bbox_parser/nvdsparsebbox_mmyolo.cpp -------------------------------------------------------------------------------- /deploy/easydeploy/deepstream/deepstream_app_config.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/deepstream/deepstream_app_config.txt -------------------------------------------------------------------------------- /deploy/easydeploy/docs/model_convert.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/docs/model_convert.md -------------------------------------------------------------------------------- /deploy/easydeploy/examples/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/examples/config.py -------------------------------------------------------------------------------- /deploy/easydeploy/examples/cv2_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/examples/cv2_nms.py -------------------------------------------------------------------------------- /deploy/easydeploy/examples/main_onnxruntime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/examples/main_onnxruntime.py -------------------------------------------------------------------------------- /deploy/easydeploy/examples/numpy_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/examples/numpy_coder.py -------------------------------------------------------------------------------- /deploy/easydeploy/examples/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/examples/preprocess.py -------------------------------------------------------------------------------- /deploy/easydeploy/examples/requirements.txt: -------------------------------------------------------------------------------- 1 | onnxruntime 2 | opencv-python==4.7.0.72 3 | -------------------------------------------------------------------------------- /deploy/easydeploy/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/model/__init__.py -------------------------------------------------------------------------------- /deploy/easydeploy/model/backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/model/backend.py -------------------------------------------------------------------------------- /deploy/easydeploy/model/backendwrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/model/backendwrapper.py -------------------------------------------------------------------------------- /deploy/easydeploy/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/model/model.py -------------------------------------------------------------------------------- /deploy/easydeploy/nms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/nms/__init__.py -------------------------------------------------------------------------------- /deploy/easydeploy/nms/ort_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/nms/ort_nms.py -------------------------------------------------------------------------------- /deploy/easydeploy/nms/trt_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/nms/trt_nms.py -------------------------------------------------------------------------------- /deploy/easydeploy/onnx_demo.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deploy/easydeploy/tools/build_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/tools/build_engine.py -------------------------------------------------------------------------------- /deploy/easydeploy/tools/export_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/tools/export_onnx.py -------------------------------------------------------------------------------- /deploy/easydeploy/tools/image-demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/easydeploy/tools/image-demo.py -------------------------------------------------------------------------------- /deploy/export_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/export_onnx.py -------------------------------------------------------------------------------- /deploy/onnx_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/onnx_demo.py -------------------------------------------------------------------------------- /deploy/tflite_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/deploy/tflite_demo.py -------------------------------------------------------------------------------- /docs/data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/docs/data.md -------------------------------------------------------------------------------- /docs/deploy.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/docs/deploy.md -------------------------------------------------------------------------------- /docs/faq.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/docs/faq.md -------------------------------------------------------------------------------- /docs/finetuning.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/docs/finetuning.md -------------------------------------------------------------------------------- /docs/installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/docs/installation.md -------------------------------------------------------------------------------- /docs/prompt_yolo_world.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/docs/prompt_yolo_world.md -------------------------------------------------------------------------------- /docs/reparameterize.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/docs/reparameterize.md -------------------------------------------------------------------------------- /docs/tflite_deploy.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/docs/tflite_deploy.md -------------------------------------------------------------------------------- /docs/update_20250123.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/docs/update_20250123.md -------------------------------------------------------------------------------- /docs/updates.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/docs/updates.md -------------------------------------------------------------------------------- /docs/yoloworld_v2_results.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/docs/yoloworld_v2_results.md -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements/basic_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/requirements/basic_requirements.txt -------------------------------------------------------------------------------- /requirements/demo_requirements.txt: -------------------------------------------------------------------------------- 1 | gradio==4.16.0 2 | supervision -------------------------------------------------------------------------------- /requirements/onnx_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/requirements/onnx_requirements.txt -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/generate_image_prompts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/tools/generate_image_prompts.py -------------------------------------------------------------------------------- /tools/generate_text_prompts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/tools/generate_text_prompts.py -------------------------------------------------------------------------------- /tools/reparameterize_yoloworld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/tools/reparameterize_yoloworld.py -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/tools/train.py -------------------------------------------------------------------------------- /yolo_world/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/__init__.py -------------------------------------------------------------------------------- /yolo_world/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/datasets/__init__.py -------------------------------------------------------------------------------- /yolo_world/datasets/mm_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/datasets/mm_dataset.py -------------------------------------------------------------------------------- /yolo_world/datasets/transformers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/datasets/transformers/__init__.py -------------------------------------------------------------------------------- /yolo_world/datasets/transformers/mm_mix_img_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/datasets/transformers/mm_mix_img_transforms.py -------------------------------------------------------------------------------- /yolo_world/datasets/transformers/mm_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/datasets/transformers/mm_transforms.py -------------------------------------------------------------------------------- /yolo_world/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/datasets/utils.py -------------------------------------------------------------------------------- /yolo_world/datasets/yolov5_cc3m_grounding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/datasets/yolov5_cc3m_grounding.py -------------------------------------------------------------------------------- /yolo_world/datasets/yolov5_lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/datasets/yolov5_lvis.py -------------------------------------------------------------------------------- /yolo_world/datasets/yolov5_mixed_grounding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/datasets/yolov5_mixed_grounding.py -------------------------------------------------------------------------------- /yolo_world/datasets/yolov5_obj365v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/datasets/yolov5_obj365v1.py -------------------------------------------------------------------------------- /yolo_world/datasets/yolov5_obj365v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/datasets/yolov5_obj365v2.py -------------------------------------------------------------------------------- /yolo_world/datasets/yolov5_v3det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/datasets/yolov5_v3det.py -------------------------------------------------------------------------------- /yolo_world/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/engine/__init__.py -------------------------------------------------------------------------------- /yolo_world/engine/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/engine/optimizers/__init__.py -------------------------------------------------------------------------------- /yolo_world/engine/optimizers/yolow_v5_optim_constructor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/engine/optimizers/yolow_v5_optim_constructor.py -------------------------------------------------------------------------------- /yolo_world/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/__init__.py -------------------------------------------------------------------------------- /yolo_world/models/assigner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/assigner/__init__.py -------------------------------------------------------------------------------- /yolo_world/models/assigner/task_aligned_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/assigner/task_aligned_assigner.py -------------------------------------------------------------------------------- /yolo_world/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/backbones/__init__.py -------------------------------------------------------------------------------- /yolo_world/models/backbones/mm_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/backbones/mm_backbone.py -------------------------------------------------------------------------------- /yolo_world/models/data_preprocessors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/data_preprocessors/__init__.py -------------------------------------------------------------------------------- /yolo_world/models/data_preprocessors/data_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/data_preprocessors/data_preprocessor.py -------------------------------------------------------------------------------- /yolo_world/models/dense_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/dense_heads/__init__.py -------------------------------------------------------------------------------- /yolo_world/models/dense_heads/yolo_world_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/dense_heads/yolo_world_head.py -------------------------------------------------------------------------------- /yolo_world/models/dense_heads/yolo_world_seg_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/dense_heads/yolo_world_seg_head.py -------------------------------------------------------------------------------- /yolo_world/models/detectors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/detectors/__init__.py -------------------------------------------------------------------------------- /yolo_world/models/detectors/yolo_world.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/detectors/yolo_world.py -------------------------------------------------------------------------------- /yolo_world/models/detectors/yolo_world_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/detectors/yolo_world_image.py -------------------------------------------------------------------------------- /yolo_world/models/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/layers/__init__.py -------------------------------------------------------------------------------- /yolo_world/models/layers/yolo_bricks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/layers/yolo_bricks.py -------------------------------------------------------------------------------- /yolo_world/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/losses/__init__.py -------------------------------------------------------------------------------- /yolo_world/models/losses/dynamic_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/losses/dynamic_loss.py -------------------------------------------------------------------------------- /yolo_world/models/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/necks/__init__.py -------------------------------------------------------------------------------- /yolo_world/models/necks/yolo_world_pafpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/models/necks/yolo_world_pafpn.py -------------------------------------------------------------------------------- /yolo_world/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/YOLO-World/HEAD/yolo_world/version.py --------------------------------------------------------------------------------