update preds

SysCV · Nov 15, 2022 · 2c18172 · 2c18172
1 parent fe4dced
commit 2c18172
Show file tree

Hide file tree

Showing 8 changed files with 15 additions and 16 deletions.
diff --git a/README.md b/README.md
@@ -50,15 +50,15 @@ Our method outperforms the states of the art on BDD100K, and TAO benchmarks.
 
 ### TAO val set
 
-| Method                                              | backbone          | pretrain     | TETA | LocA | AssocA | ClsA | config                               | model                                                                                              |
-|-----------------------------------------------------|-------------------|--------------|------|------|--------|------|--------------------------------------|----------------------------------------------------------------------------------------------------|
-| [QDTrack(CVPR21)](https://arxiv.org/abs/2006.06664) | ResNet-101        | ImageNet-1K  | 30.0 | 50.5 | 27.4   | 12.1 | -                                    | -                                                                                                  |
-| TETer (Ours)                                        | ResNet-101        | ImageNet-1K  | 33.3 | 51.6 | 35.0   | 13.2 | [cfg](configs/tao/cem_r101_lvis.py)  | [google drive](https://drive.google.com/file/d/17koyuCbnj42ioZRxZZ5DChmCoAaMwets/view?usp=sharing) |
-| TETer-HTC (Ours)                                    | ResNeXt-101-64x4d | ImageNet-1K  | 36.9 | 57.5 | 37.5   | 15.7 | -                                    | -                                                                                                  |
-| TETer-SwinT (Ours)                                  | SwinT             | ImageNet-1K  | 34.6 | 52.1 | 36.7   | 15.0 | [cfg](configs/tao/cem_swinT_lvis.py) | [google drive](https://drive.google.com/file/d/1UvMkH1xfCqNghiDCbb6jE_9yklYggmlM/view?usp=sharing) |
-| TETer-SwinS (Ours)                                  | SwinS             | ImageNet-1K  | 36.7 | 54.2 | 38.4   | 17.4 | [cfg](configs/tao/cem_swinS_lvis.py) | [google drive](https://drive.google.com/file/d/1jRjrj57PKi7fLKBgoq2Dt5SZJC4Nejip/view?usp=sharing) |
-| TETer-SwinB (Ours)                                  | SwinB             | ImageNet-22K | 38.8 | 55.6 | 40.1   | 20.8 | [cfg](configs/tao/cem_swinB_lvis.py) | [google drive](https://drive.google.com/file/d/11Y3Vv1gB4m0Vn7AKI5XW1TIsCOq30hKB/view?usp=sharing) |
-| TETer-SwinL (Ours)                                  | SwinL             | ImageNet-22K | 40.1 | 56.3 | 39.9   | 24.1 | [cfg](configs/tao/cem_swinL_lvis.py) | [google drive](https://drive.google.com/file/d/1J3z5pbF7trU3fDCVfizAPsmp3oV9NVj8/view?usp=sharing)                                                                                   |
+| Method                                              | backbone          | pretrain     | TETA | LocA | AssocA | ClsA | config                               | model                                                                                              | preds                                                                                                 |
+|-----------------------------------------------------|-------------------|--------------|------|------|--------|------|--------------------------------------|----------------------------------------------------------------------------------------------------|-------------------------------------------------------------------------------------------------------|
+| [QDTrack(CVPR21)](https://arxiv.org/abs/2006.06664) | ResNet-101        | ImageNet-1K  | 30.0 | 50.5 | 27.4   | 12.1 | -                                    | -                                                                                                  | -                                                                                                     |
+| TETer (Ours)                                        | ResNet-101        | ImageNet-1K  | 33.3 | 51.6 | 35.0   | 13.2 | [cfg](configs/tao/cem_r101_lvis.py)  | [google drive](https://drive.google.com/file/d/17koyuCbnj42ioZRxZZ5DChmCoAaMwets/view?usp=sharing) | coming soon                                                                                           |
+| TETer-HTC (Ours)                                    | ResNeXt-101-64x4d | ImageNet-1K  | 36.9 | 57.5 | 37.5   | 15.7 | -                                    | -                                                                                                  | -                                                                                                     |
+| TETer-SwinT (Ours)                                  | SwinT             | ImageNet-1K  | 34.6 | 52.1 | 36.7   | 15.0 | [cfg](configs/tao/cem_swinT_lvis.py) | [google drive](https://drive.google.com/file/d/1UvMkH1xfCqNghiDCbb6jE_9yklYggmlM/view?usp=sharing) | coming soon                                                                                           |
+| TETer-SwinS (Ours)                                  | SwinS             | ImageNet-1K  | 36.7 | 54.2 | 38.4   | 17.4 | [cfg](configs/tao/cem_swinS_lvis.py) | [google drive](https://drive.google.com/file/d/1jRjrj57PKi7fLKBgoq2Dt5SZJC4Nejip/view?usp=sharing) | coming soon                                                                                           |
+| TETer-SwinB (Ours)                                  | SwinB             | ImageNet-22K | 38.8 | 55.6 | 40.1   | 20.8 | [cfg](configs/tao/cem_swinB_lvis.py) | [google drive](https://drive.google.com/file/d/11Y3Vv1gB4m0Vn7AKI5XW1TIsCOq30hKB/view?usp=sharing) | coming soon                                                                                           |
+| TETer-SwinL (Ours)                                  | SwinL             | ImageNet-22K | 40.1 | 56.3 | 39.9   | 24.1 | [cfg](configs/tao/cem_swinL_lvis.py) | [google drive](https://drive.google.com/file/d/1J3z5pbF7trU3fDCVfizAPsmp3oV9NVj8/view?usp=sharing) | [google drive](https://drive.google.com/file/d/1ZXXV-hQOxC-luSQcH3ph1k1ETUe6rYKV/view?usp=share_link) |    
 
 ## Installation
 

diff --git a/configs/tao/cem_swinT_lvis.py b/configs/tao/cem_swinT_lvis.py
@@ -43,7 +43,6 @@
                             pos_normalize=True,
                             loss_weight=0.25)
             , softmax_temp=-1),
-
         track_head=dict(
             type='QuasiDenseEmbedHead',
             num_convs=4,

diff --git a/teter/datasets/coco_video_dataset.py b/teter/datasets/coco_video_dataset.py
@@ -7,7 +7,7 @@
 from .parsers import CocoVID
 
 
-@DATASETS.register_module()
+@DATASETS.register_module(force=True)
 class CocoVideoDataset(CocoDataset):
 
     CLASSES = None

diff --git a/teter/datasets/tao_dataset.py b/teter/datasets/tao_dataset.py
@@ -36,7 +36,7 @@ def majority_vote(prediction):
     return class_by_majority_count_res
 
 
-@DATASETS.register_module()
+@DATASETS.register_module(force=True)
 class TaoDataset(CocoVideoDataset):
     def load_annotations(self, ann_file):
         """Load annotation from annotation file."""

diff --git a/teter/models/losses/l2_loss.py b/teter/models/losses/l2_loss.py
@@ -20,7 +20,7 @@ def l2_loss(pred, target):
     return loss
 
 
-@LOSSES.register_module()
+@LOSSES.register_module(force=True)
 class L2Loss(nn.Module):
     """L2 loss.
 

diff --git a/teter/models/losses/multipos_cross_entropy_loss.py b/teter/models/losses/multipos_cross_entropy_loss.py
@@ -60,7 +60,7 @@ def multi_pos_cross_entropy(
     return loss
 
 
-@LOSSES.register_module()
+@LOSSES.register_module(force=True)
 class MultiPosCrossEntropyLoss(nn.Module):
     def __init__(self, reduction="mean", loss_weight=1.0, version="v3"):
         super(MultiPosCrossEntropyLoss, self).__init__()

diff --git a/teter/models/roi_heads/track_heads/cem_head.py b/teter/models/roi_heads/track_heads/cem_head.py
@@ -7,7 +7,7 @@
 from teter.core import cal_similarity
 
 
-@HEADS.register_module()
+@HEADS.register_module(force=True)
 class ClsExemplarHead(nn.Module):
     def __init__(
         self,

diff --git a/teter/models/roi_heads/track_heads/quasi_dense_embed_head.py b/teter/models/roi_heads/track_heads/quasi_dense_embed_head.py
@@ -7,7 +7,7 @@
 from teter.core import cal_similarity
 
 
-@HEADS.register_module()
+@HEADS.register_module(force=True)
 class QuasiDenseEmbedHead(nn.Module):
     def __init__(
         self,