From 2c18172607d6fc526083c67d93e80104fdc98577 Mon Sep 17 00:00:00 2001 From: siyliepfl Date: Tue, 15 Nov 2022 17:13:40 +0100 Subject: [PATCH] update preds --- README.md | 18 +++++++++--------- configs/tao/cem_swinT_lvis.py | 1 - teter/datasets/coco_video_dataset.py | 2 +- teter/datasets/tao_dataset.py | 2 +- teter/models/losses/l2_loss.py | 2 +- .../losses/multipos_cross_entropy_loss.py | 2 +- teter/models/roi_heads/track_heads/cem_head.py | 2 +- .../track_heads/quasi_dense_embed_head.py | 2 +- 8 files changed, 15 insertions(+), 16 deletions(-) diff --git a/README.md b/README.md index 79cb0e0..bf7d50a 100644 --- 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 index 144cab0..1c8fbf4 100644 --- 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 index f318f65..b3b2f9e 100644 --- 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 index 3311ae9..150c0f1 100644 --- 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 index b2b675e..5ad4e97 100644 --- 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 index 2701855..de65a0a 100644 --- 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 index fba29f9..afb69bd 100644 --- 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 index cfd750e..ecfafef 100644 --- 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,