add odinw configs and evaluation results of GLIP(#11175)

open-mmlab · Nov 15, 2023 · 24bb129 · 24bb129
1 parent 5a02a0a
commit 24bb129
Show file tree

Hide file tree

Showing 6 changed files with 153 additions and 7 deletions.
diff --git a/configs/glip/README.md b/configs/glip/README.md
@@ -92,7 +92,7 @@ Note:
 |   GLIP-T   |    ✔     |             |             |             |            |            |            |            |           |    O365,GoldG,CC3M,SBU     | [config](lvis/glip_atss_swin-t_bc_fpn_dyhead_pretrain_zeroshot_lvis.py) |  [model](https://download.openmmlab.com/mmdetection/v3.0/glip/glip_tiny_mmdet-c24ce662.pth)  |
 |   GLIP-T   |          |    18.1     |    21.2     |    33.1     |    26.7    |    10.8    |    14.7    |    29.0    |   19.6    |    O365,GoldG,CC3M,SBU     | [config](lvis/glip_atss_swin-t_bc_fpn_dyhead_pretrain_zeroshot_lvis.py) |  [model](https://download.openmmlab.com/mmdetection/v3.0/glip/glip_tiny_mmdet-c24ce662.pth)  |
 |   GLIP-L   |    ✔     |    29.2     |    34.9     |    42.1     |    37.9    |            |            |            |           | FourODs,GoldG,CC3M+12M,SBU |  [config](lvis/glip_atss_swin-l_fpn_dyhead_pretrain_zeroshot_lvis.py)   |   [model](https://download.openmmlab.com/mmdetection/v3.0/glip/glip_l_mmdet-abfe026b.pth)    |
-|   GLIP-L   |          |    27.9     |    33.7     |    39.7     |    36.1    |            |            |            |           | FourODs,GoldG,CC3M+12M,SBU |  [config](lvis/glip_atss_swin-l_fpn_dyhead_pretrain_zeroshot_lvis.py)   |   [model](https://download.openmmlab.com/mmdetection/v3.0/glip/glip_l_mmdet-abfe026b.pth)    |
+|   GLIP-L   |          |    27.9     |    33.7     |    39.7     |    36.1    |    20.2    |    25.8    |    35.3    |   28.5    | FourODs,GoldG,CC3M+12M,SBU |  [config](lvis/glip_atss_swin-l_fpn_dyhead_pretrain_zeroshot_lvis.py)   |   [model](https://download.openmmlab.com/mmdetection/v3.0/glip/glip_l_mmdet-abfe026b.pth)    |
 
 Note:
 

diff --git a/configs/odinw/README.md b/configs/odinw/README.md
@@ -0,0 +1,104 @@
+# ODinW
+
+[ELEVATER: A Benchmark and Toolkit for Evaluating Language-Augmented Visual Models](https://arxiv.org/pdf/2204.08790.pdf)
+
+<!-- [DATASET] -->
+
+## Get Started
+
+1. development Developmennt Setup can reger to hits /\\To download dataset, you can refer to [reference document](../../docs/zh_cn/user_guides/dataset_prepare.md)
+
+2. You can use the following data to run the inference.
+
+```shell
+cd $MMDETROOT
+
+python tools/test.py configs/odinw35/glip_atss_swin_t_fpn_dataset.py checkpoints/glip_tiny_a_mmdet-b3654169.pth
+```
+
+## Abstract
+
+Learning visual representations from natural language supervision has recently shown great promise in a number of pioneering works. In general, these language-augmented visual models demonstrate strong transferability to a variety of datasets and tasks. However, it remains challenging to evaluate the transferablity of these models due to the lack of easy-to-use evaluation toolkits and public benchmarks. To tackle this, we build ELEVATER 1 , the first benchmark and toolkit for evaluating (pre-trained) language-augmented visual models. ELEVATER is composed of three components. (i) Datasets. As downstream evaluation suites, it consists of 20 image classification datasets and 35 object detection datasets, each of which is augmented with external knowledge. (ii) Toolkit. An automatic hyper-parameter tuning toolkit is developed to facilitate model evaluation on downstream tasks. (iii) Metrics. A variety of evaluation metrics are used to measure sample-efficiency (zero-shot and few-shot) and parameter-efficiency (linear probing and full model fine-tuning). ELEVATER is platform for Computer Vision in the Wild (CVinW), and is publicly released at https://computer-vision-in-the-wild.github.io/ELEVATER/
+
+## Results and models of odinw13
+
+| Method                | GLIP-T(A) | Official  | GLIP-T(B) | Official  | GLIP-T(C) | Official  |
+| --------------------- | --------- | --------- | --------- | --------- | --------- | --------- |
+| AerialMaritimeDrone   | 0.123     | 0.122     | 0.110     | 0.11      | 0.130     | 0.130     |
+| Aquarium              | 0.175     | 0.174     | 0.173     | 0.169     | 0.191     | 0.190     |
+| CottontailRabbits     | 0.686     | 0.686     | 0.688     | 0.688     | 0.744     | 0.744     |
+| EgoHands              | 0.013     | 0.013     | 0.003     | 0.540     | 0.314     | 0.315     |
+| NorthAmericaMushrooms | 0.502     | 0.502     | 0.367     | 0.051     | 0.297     | 0.296     |
+| Packages              | 0.589     | 0.589     | 0.083     | 0.030     | 0.699     | 0.699     |
+| PascalVOC             | 0.512     | 0.512     | 0.541     | 0.288     | 0.565     | 0.565     |
+| pistols               | 0.339     | 0.339     | 0.502     | 0.338     | 0.503     | 0.504     |
+| pothole               | 0.007     | 0.007     | 0.030     | 0.475     | 0.058     | 0.058     |
+| Raccoon               | 0.075     | 0.075     | 0.285     | 0.288     | 0.241     | 0.244     |
+| ShellfishOpenImages   | 0.372     | 0.372     | 0.337     | 0.338     | 0.300     | 0.302     |
+| thermalDogsAndPeople  | 0.372     | 0.372     | 0.475     | 0.475     | 0.510     | 0.510     |
+| VehiclesOpenImages    | 0.574     | 0.574     | 0.562     | 0.547     | 0.549     | 0.534     |
+| Average               | **0.334** | **0.324** | **0.320** | **0.318** | **0.392** | **0.392** |
+
+Note:
+
+1. The above are zero-shot evaluation results.
+2. The config and weights can be found at [here](../glip/README.md)
+
+## Results and models of odinw35
+
+| Method                      | GLIP-T(A) | Official  | GLIP-T(B) | Official  | GLIP-T(C) | Official  |
+| --------------------------- | --------- | --------- | --------- | --------- | --------- | --------- |
+| AerialMaritimeDrone_large   | 0.123     | 0.122     | 0.110     | 0.110     | 0.130     | 0.130     |
+| AerialMaritimeDrone_tiled   | 0.174     | 0.174     | 0.172     | 0.172     | 0.172     | 0.172     |
+| AmericanSignLanguageLetters | 0.001     | 0.001     | 0.003     | 0.003     | 0.009     | 0.009     |
+| Aquarium                    | 0.175     | 0.175     | 0.173     | 0.171     | 0.192     | 0.182     |
+| BCCD                        | 0.016     | 0.016     | 0.001     | 0.001     | 0.000     | 0.000     |
+| boggleBoards                | 0.000     | 0.000     | 0.000     | 0.000     | 0.000     | 0.000     |
+| brackishUnderwater          | 0.016     | 0..013    | 0.021     | 0.027     | 0.020     | 0.022     |
+| ChessPieces                 | 0.001     | 0.001     | 0.000     | 0.000     | 0.001     | 0.001     |
+| CottontailRabbits           | 0.710     | 0.709     | 0.683     | 0.683     | 0.752     | 0.752     |
+| dice                        | 0.005     | 0.005     | 0.004     | 0.004     | 0.004     | 0.004     |
+| DroneControl                | 0.016     | 0.017     | 0.006     | 0.008     | 0.005     | 0.007     |
+| EgoHands_generic            | 0.009     | 0.010     | 0.005     | 0.006     | 0.510     | 0.508     |
+| EgoHands_specific           | 0.001     | 0.001     | 0.004     | 0.006     | 0.003     | 0.004     |
+| HardHatWorkers              | 0.029     | 0.029     | 0.023     | 0.023     | 0.033     | 0.033     |
+| MaskWearing                 | 0.007     | 0.007     | 0.003     | 0.002     | 0.005     | 0.005     |
+| MountainDewCommercial       | 0.218     | 0.227     | 0.199     | 0.197     | 0.478     | 0.463     |
+| NorthAmericaMushrooms       | 0.502     | 0.502     | 0.450     | 0.450     | 0.497     | 0.497     |
+| openPoetryVision            | 0.000     | 0.000     | 0.000     | 0.000     | 0.000     | 0.000     |
+| OxfordPets_by_breed         | 0.001     | 0.002     | 0.002     | 0.004     | 0.001     | 0.002     |
+| OxfordPets_by_species       | 0.016     | 0.011     | 0.012     | 0.009     | 0.013     | 0.009     |
+| PKLot                       | 0.002     | 0.002     | 0.000     | 0.000     | 0.000     | 0.000     |
+| Packages                    | 0.569     | 0.569     | 0.279     | 0.279     | 0.712     | 0.712     |
+| PascalVOC                   | 0.512     | 0.512     | 0.541     | 0.540     | 0.565     | 0.565     |
+| pistols                     | 0.339     | 0.339     | 0.502     | 0.501     | 0.503     | 0.504     |
+| plantdoc                    | 0.002     | 0.002     | 0.007     | 0.007     | 0.009     | 0.009     |
+| pothole                     | 0.007     | 0.010     | 0.024     | 0.025     | 0.085     | 0.101     |
+| Raccoons                    | 0.075     | 0.074     | 0.285     | 0.288     | 0.241     | 0.244     |
+| selfdrivingCar              | 0.071     | 0.072     | 0.074     | 0.074     | 0.081     | 0.080     |
+| ShellfishOpenImages         | 0.253     | 0.253     | 0.337     | 0.338     | 0.300     | 0.302     |
+| ThermalCheetah              | 0.028     | 0.028     | 0.000     | 0.000     | 0.028     | 0.028     |
+| thermalDogsAndPeople        | 0.372     | 0.372     | 0.475     | 0.475     | 0.510     | 0.510     |
+| UnoCards                    | 0.000     | 0.000     | 0.000     | 0.001     | 0.002     | 0.003     |
+| VehiclesOpenImages          | 0.574     | 0.566     | 0.562     | 0.547     | 0.549     | 0.534     |
+| WildfireSmoke               | 0.000     | 0.000     | 0.000     | 0.000     | 0.017     | 0.017     |
+| websiteScreenshots          | 0.003     | 0.004     | 0.003     | 0.005     | 0.005     | 0.006     |
+| Average                     | **0.134** | **0.134** | **0.138** | **0.138** | **0.179** | **0.178** |
+
+Note:
+
+1. The above are zero-shot evaluation results.
+2. The config and weights can be found at [here](../glip/README.md)
+
+## Citation
+
+```
+@misc{li2022elevater,
+      title={ELEVATER: A Benchmark and Toolkit for Evaluating Language-Augmented Visual Models},
+      author={Chunyuan Li and Haotian Liu and Liunian Harold Li and Pengchuan Zhang and Jyoti Aneja and Jianwei Yang and Ping Jin and Houdong Hu and Zicheng Liu and Yong Jae Lee and Jianfeng Gao},
+      year={2022},
+      eprint={2204.08790},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV}
+}
+```
diff --git a/configs/odinw/glip_atss_swin-t_a_fpn_dyhead_pretrain_odinw35.py b/configs/odinw/glip_atss_swin-t_a_fpn_dyhead_pretrain_odinw35.py
@@ -3,6 +3,11 @@
 dataset_type = 'CocoDataset'
 data_root = 'data/odinw/'
 
+base_test_pipeline = _base_.test_pipeline
+base_test_pipeline[-1]['meta_keys'] = ('img_id', 'img_path', 'ori_shape',
+                                       'img_shape', 'scale_factor', 'text',
+                                       'custom_entities', 'caption_prompt')
+
 # ---------------------1 AerialMaritimeDrone_large---------------------#
 class_name = ('boat', 'car', 'dock', 'jetski', 'lift')
 metainfo = dict(classes=class_name)
@@ -154,21 +159,21 @@
     metric='bbox')
 
 # ---------------------9 CottontailRabbits---------------------#
-class_name = ('Cottontail-Rabbit', )
+class_name = ('rabbit', )
 metainfo = dict(classes=class_name)
 _data_root = data_root + 'CottontailRabbits/'
 dataset_CottontailRabbits = dict(
     type=dataset_type,
     metainfo=metainfo,
     data_root=_data_root,
-    ann_file='valid/annotations_without_background.json',
+    ann_file='valid/new_annotations_without_background.json',
     data_prefix=dict(img='valid/'),
     pipeline=_base_.test_pipeline,
     test_mode=True,
     return_classes=True)
 val_evaluator_CottontailRabbits = dict(
     type='CocoMetric',
-    ann_file=_data_root + 'valid/annotations_without_background.json',
+    ann_file=_data_root + 'valid/new_annotations_without_background.json',
     metric='bbox')
 
 # ---------------------10 dice---------------------#
@@ -212,13 +217,15 @@
 class_name = ('hand', )
 metainfo = dict(classes=class_name)
 _data_root = data_root + 'EgoHands/generic/'
+caption_prompt = {'hand': {'suffix': ' of a person'}}
 dataset_EgoHands_generic = dict(
     type=dataset_type,
     metainfo=metainfo,
     data_root=_data_root,
     ann_file='valid/annotations_without_background.json',
     data_prefix=dict(img='valid/'),
-    pipeline=_base_.test_pipeline,
+    pipeline=base_test_pipeline,
+    caption_prompt=caption_prompt,
     test_mode=True,
     return_classes=True)
 val_evaluator_EgoHands_generic = dict(
@@ -414,13 +421,20 @@
 class_name = ('package', )
 metainfo = dict(classes=class_name)
 _data_root = data_root + 'Packages/Raw/'
+caption_prompt = {
+    'package': {
+        'prefix': 'there is a ',
+        'suffix': ' on the porch'
+    }
+}
 dataset_Packages = dict(
     type=dataset_type,
     metainfo=metainfo,
     data_root=_data_root,
     ann_file='valid/annotations_without_background.json',
     data_prefix=dict(img='valid/'),
-    pipeline=_base_.test_pipeline,
+    pipeline=base_test_pipeline,
+    caption_prompt=caption_prompt,
     test_mode=True,
     return_classes=True)
 val_evaluator_Packages = dict(
@@ -501,13 +515,21 @@
 class_name = ('pothole', )
 metainfo = dict(classes=class_name)
 _data_root = data_root + 'pothole/'
+caption_prompt = {
+    'pothole': {
+        'name': 'holes',
+        'prefix': 'there are some',
+        'suffix': ' on the road'
+    }
+}
 dataset_pothole = dict(
     type=dataset_type,
     metainfo=metainfo,
     data_root=_data_root,
     ann_file='valid/annotations_without_background.json',
     data_prefix=dict(img='valid/'),
-    pipeline=_base_.test_pipeline,
+    caption_prompt=caption_prompt,
+    pipeline=base_test_pipeline,
     test_mode=True,
     return_classes=True)
 val_evaluator_pothole = dict(
@@ -683,6 +705,7 @@
     metric='bbox')
 
 # --------------------- Config---------------------#
+
 dataset_prefixes = [
     'AerialMaritimeDrone_large',
     'AerialMaritimeDrone_tiled',

diff --git a/configs/odinw/glip_atss_swin-t_bc_fpn_dyhead_pretrain_odinw13.py b/configs/odinw/glip_atss_swin-t_bc_fpn_dyhead_pretrain_odinw13.py
@@ -0,0 +1,3 @@
+_base_ = './glip_atss_swin-t_a_fpn_dyhead_pretrain_odinw13.py'
+
+model = dict(bbox_head=dict(early_fuse=True))
diff --git a/configs/odinw/glip_atss_swin-t_bc_fpn_dyhead_pretrain_odinw35.py b/configs/odinw/glip_atss_swin-t_bc_fpn_dyhead_pretrain_odinw35.py
@@ -0,0 +1,3 @@
+_base_ = './glip_atss_swin-t_a_fpn_dyhead_pretrain_odinw35.py'
+
+model = dict(bbox_head=dict(early_fuse=True))
diff --git a/configs/odinw/override_category.py b/configs/odinw/override_category.py
@@ -74,6 +74,19 @@ def main():
     mmengine.dump(json_data,
                   _data_root + 'valid/new_annotations_without_background.json')
 
+    CottontailRabbits = [{
+        'id': 1,
+        'name': 'rabbit',
+        'supercategory': 'Cottontail-Rabbit'
+    }]
+
+    _data_root = args.data_root + 'CottontailRabbits/'
+    json_data = mmengine.load(_data_root +
+                              'valid/annotations_without_background.json')
+    json_data['categories'] = CottontailRabbits
+    mmengine.dump(json_data,
+                  _data_root + 'valid/new_annotations_without_background.json')
+
     NorthAmericaMushrooms = [{
         'id': 1,
         'name': 'flat mushroom',