adjust commit

Liqiu-Chen · Jul 20, 2024 · 4c9d5d0 · 4c9d5d0
1 parent 10bba86
commit 4c9d5d0
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/.idea/vcs.xml b/.idea/vcs.xml
diff --git a/lib/models/tbsi_track/tbsi_track.py b/lib/models/tbsi_track/tbsi_track.py
@@ -11,11 +11,11 @@
 from torch.nn.modules.transformer import _get_clones
 
 from lib.models.layers.head import build_box_head, conv
-from lib.models.tbsi_track.vit_tbsi_care import vit_base_patch16_224_tbsi
+from lib.models.tbsi_track.vit_iimf_care import vit_base_patch16_224_tbsi
 from lib.utils.box_ops import box_xyxy_to_cxcywh
 
 
-class TBSITrack(nn.Module):
+class IIMFTrack(nn.Module):
     """ This is the base class for TBSITrack developed on OSTrack (Ye et al. ECCV 2022) """
 
     def __init__(self, transformer, box_head, aux_loss=False, head_type="CORNER"):
@@ -139,7 +139,7 @@ def build_tbsi_track(cfg, training=True):
 
     box_head = build_box_head(cfg, hidden_dim)
 
-    model = TBSITrack(
+    model = IIMFTrack(
         backbone,
         box_head,
         aux_loss=False,

diff --git a/lib/models/tbsi_track/vit_tbsi_care.py → lib/models/tbsi_track/vit_iimf_care.py b/lib/models/tbsi_track/vit_tbsi_care.py → lib/models/tbsi_track/vit_iimf_care.py