fix typo

shimomurakei · Mar 1, 2024 · 860500f · 860500f
1 parent 9764e53
commit 860500f
Show file tree

Hide file tree

Showing 13 changed files with 14 additions and 14 deletions.
diff --git a/configs/finetune_coco/yolo_world_l_dual_vlpan_2e-4_80e_8gpus_finetune_coco.py b/configs/finetune_coco/yolo_world_l_dual_vlpan_2e-4_80e_8gpus_finetune_coco.py
@@ -35,7 +35,7 @@
             type='HuggingCLIPLanguageBackbone',
             model_name='openai/clip-vit-base-patch32',
             frozen_modules=['all'])),
-    neck=dict(type='YOLOWolrdDualPAFPN',
+    neck=dict(type='YOLOWorldDualPAFPN',
               guide_channels=text_channels,
               embed_channels=neck_embed_channels,
               num_heads=neck_num_heads,

diff --git a/configs/finetune_coco/yolo_world_l_dual_vlpan_2e-4_80e_8gpus_mask-refine_finetune_coco.py b/configs/finetune_coco/yolo_world_l_dual_vlpan_2e-4_80e_8gpus_mask-refine_finetune_coco.py
@@ -35,7 +35,7 @@
             type='HuggingCLIPLanguageBackbone',
             model_name='openai/clip-vit-base-patch32',
             frozen_modules=['all'])),
-    neck=dict(type='YOLOWolrdDualPAFPN',
+    neck=dict(type='YOLOWorldDualPAFPN',
               guide_channels=text_channels,
               embed_channels=neck_embed_channels,
               num_heads=neck_num_heads,

diff --git a/...ain/yolo_world_l_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py b/...ain/yolo_world_l_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py
@@ -31,7 +31,7 @@
             type='HuggingCLIPLanguageBackbone',
             model_name='openai/clip-vit-base-patch32',
             frozen_modules=['all'])),
-    neck=dict(type='YOLOWolrdDualPAFPN',
+    neck=dict(type='YOLOWorldDualPAFPN',
               guide_channels=text_channels,
               embed_channels=neck_embed_channels,
               num_heads=neck_num_heads,

diff --git a/...retrain/yolo_world_l_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_val.py b/...retrain/yolo_world_l_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_val.py
@@ -31,7 +31,7 @@
             type='HuggingCLIPLanguageBackbone',
             model_name='openai/clip-vit-base-patch32',
             frozen_modules=['all'])),
-    neck=dict(type='YOLOWolrdDualPAFPN',
+    neck=dict(type='YOLOWorldDualPAFPN',
               guide_channels=text_channels,
               embed_channels=neck_embed_channels,
               num_heads=neck_num_heads,

diff --git a/...ain/yolo_world_m_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py b/...ain/yolo_world_m_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py
@@ -31,7 +31,7 @@
             type='HuggingCLIPLanguageBackbone',
             model_name='openai/clip-vit-base-patch32',
             frozen_modules=['all'])),
-    neck=dict(type='YOLOWolrdDualPAFPN',
+    neck=dict(type='YOLOWorldDualPAFPN',
               guide_channels=text_channels,
               embed_channels=neck_embed_channels,
               num_heads=neck_num_heads,

diff --git a/...ain/yolo_world_s_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py b/...ain/yolo_world_s_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py
@@ -31,7 +31,7 @@
             type='HuggingCLIPLanguageBackbone',
             model_name='openai/clip-vit-base-patch32',
             frozen_modules=['all'])),
-    neck=dict(type='YOLOWolrdDualPAFPN',
+    neck=dict(type='YOLOWorldDualPAFPN',
               guide_channels=text_channels,
               embed_channels=neck_embed_channels,
               num_heads=neck_num_heads,

diff --git a/...ain/yolo_world_x_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py b/...ain/yolo_world_x_dual_vlpan_l2norm_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py
@@ -31,7 +31,7 @@
             type='HuggingCLIPLanguageBackbone',
             model_name='openai/clip-vit-base-patch32',
             frozen_modules=['all'])),
-    neck=dict(type='YOLOWolrdDualPAFPN',
+    neck=dict(type='YOLOWorldDualPAFPN',
               guide_channels=text_channels,
               embed_channels=neck_embed_channels,
               num_heads=neck_num_heads,

diff --git a/configs/segmentation/yolo_world_seg_l_dual_vlpan_2e-4_80e_8gpus_allmodules_finetune_lvis.py b/configs/segmentation/yolo_world_seg_l_dual_vlpan_2e-4_80e_8gpus_allmodules_finetune_lvis.py
@@ -40,7 +40,7 @@
             type='HuggingCLIPLanguageBackbone',
             model_name='openai/clip-vit-base-patch32',
             frozen_modules=[])),
-    neck=dict(type='YOLOWolrdDualPAFPN',
+    neck=dict(type='YOLOWorldDualPAFPN',
               guide_channels=text_channels,
               embed_channels=neck_embed_channels,
               num_heads=neck_num_heads,

diff --git a/configs/segmentation/yolo_world_seg_l_dual_vlpan_2e-4_80e_8gpus_seghead_finetune_lvis.py b/configs/segmentation/yolo_world_seg_l_dual_vlpan_2e-4_80e_8gpus_seghead_finetune_lvis.py
@@ -41,7 +41,7 @@
             type='HuggingCLIPLanguageBackbone',
             model_name='openai/clip-vit-base-patch32',
             frozen_modules=['all'])),
-    neck=dict(type='YOLOWolrdDualPAFPN',
+    neck=dict(type='YOLOWorldDualPAFPN',
               freeze_all=True,
               guide_channels=text_channels,
               embed_channels=neck_embed_channels,

diff --git a/configs/segmentation/yolo_world_seg_m_dual_vlpan_2e-4_80e_8gpus_allmodules_finetune_lvis.py b/configs/segmentation/yolo_world_seg_m_dual_vlpan_2e-4_80e_8gpus_allmodules_finetune_lvis.py
@@ -40,7 +40,7 @@
             type='HuggingCLIPLanguageBackbone',
             model_name='openai/clip-vit-base-patch32',
             frozen_modules=[])),
-    neck=dict(type='YOLOWolrdDualPAFPN',
+    neck=dict(type='YOLOWorldDualPAFPN',
               guide_channels=text_channels,
               embed_channels=neck_embed_channels,
               num_heads=neck_num_heads,

diff --git a/configs/segmentation/yolo_world_seg_m_dual_vlpan_2e-4_80e_8gpus_seghead_finetune_lvis.py b/configs/segmentation/yolo_world_seg_m_dual_vlpan_2e-4_80e_8gpus_seghead_finetune_lvis.py
@@ -41,7 +41,7 @@
             type='HuggingCLIPLanguageBackbone',
             model_name='openai/clip-vit-base-patch32',
             frozen_modules=['all'])),
-    neck=dict(type='YOLOWolrdDualPAFPN',
+    neck=dict(type='YOLOWorldDualPAFPN',
               freeze_all=True,
               guide_channels=text_channels,
               embed_channels=neck_embed_channels,

diff --git a/yolo_world/models/necks/__init__.py b/yolo_world/models/necks/__init__.py
@@ -1,4 +1,4 @@
 # Copyright (c) Tencent Inc. All rights reserved.
-from .yolo_world_pafpn import YOLOWorldPAFPN, YOLOWolrdDualPAFPN
+from .yolo_world_pafpn import YOLOWorldPAFPN, YOLOWorldDualPAFPN
 
-__all__ = ['YOLOWorldPAFPN', 'YOLOWolrdDualPAFPN']
+__all__ = ['YOLOWorldPAFPN', 'YOLOWorldDualPAFPN']
diff --git a/yolo_world/models/necks/yolo_world_pafpn.py b/yolo_world/models/necks/yolo_world_pafpn.py
@@ -147,7 +147,7 @@ def forward(self, img_feats: List[Tensor], txt_feats: Tensor) -> tuple:
 
 
 @MODELS.register_module()
-class YOLOWolrdDualPAFPN(YOLOWorldPAFPN):
+class YOLOWorldDualPAFPN(YOLOWorldPAFPN):
     """Path Aggregation Network used in YOLO World v8."""
     def __init__(self,
                  in_channels: List[int],