Skip to content

Commit

Permalink
[Fix] Update links in readme of DETRs and MaskFormers (#9634)
Browse files Browse the repository at this point in the history
  • Loading branch information
Li-Qingyun authored and jshilong committed Jan 19, 2023
1 parent fbcb8a9 commit 4c11117
Show file tree
Hide file tree
Showing 18 changed files with 62 additions and 48 deletions.
6 changes: 3 additions & 3 deletions configs/conditional_detr/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -23,9 +23,9 @@ Our conditional DETR learns a conditional spatial query from the decoder embeddi

We provide the config files and models for Conditional DETR: [Conditional DETR for Fast Training Convergence](https://arxiv.org/abs/2108.06152).

| Backbone | Model | Lr schd | Mem (GB) | Inf time (fps) | box AP | Config | Download |
| :------: | :--------------: | :-----: | :------: | :------------: | :----: | :-----------------------------------------------: | :----------------------------------: |
| R-50 | Conditional DETR | 50e | 7.9 | | 40.9 | [config](./conditional_detr_r50_8xb2-50e_coco.py) | \[model\](# TODO) \| \[log\](# TODO) |
| Backbone | Model | Lr schd | Mem (GB) | Inf time (fps) | box AP | Config | Download |
| :------: | :--------------: | :-----: | :------: | :------------: | :----: | :-----------------------------------------------: | :----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
| R-50 | Conditional DETR | 50e | | | 40.9 | [config](./conditional-detr_r50_8xb2-50e_coco.py) | [model](https://download.openmmlab.com/mmdetection/v3.0/conditional_detr/conditional-detr_r50_8xb2-50e_coco/conditional-detr_r50_8xb2-50e_coco_20221121_180202-c83a1dc0.pth) \| [log](https://download.openmmlab.com/mmdetection/v3.0/conditional_detr/conditional-detr_r50_8xb2-50e_coco/conditional-detr_r50_8xb2-50e_coco_20221121_180202.log.json) |

## Citation

Expand Down
6 changes: 3 additions & 3 deletions configs/dab_detr/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -22,9 +22,9 @@ We present in this paper a novel query formulation using dynamic anchor boxes fo

We provide the config files and models for DAB-DETR: [DAB-DETR: Dynamic Anchor Boxes are Better Queries for DETR](https://arxiv.org/abs/2201.12329).

| Backbone | Model | Lr schd | Mem (GB) | Inf time (fps) | box AP | Config | Download |
| :------: | :------: | :-----: | :------: | :------------: | :----: | :---------------------------------------: | :----------------------------------: |
| R-50 | DAB-DETR | 50e | 6.4 | | 42.3 | [config](./dab-detr_r50_8xb2-50e_coco.py) | \[model\](# TODO) \| \[log\](# TODO) |
| Backbone | Model | Lr schd | Mem (GB) | Inf time (fps) | box AP | Config | Download |
| :------: | :------: | :-----: | :------: | :------------: | :----: | :---------------------------------------: | :----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
| R-50 | DAB-DETR | 50e | | | 42.3 | [config](./dab-detr_r50_8xb2-50e_coco.py) | [model](https://download.openmmlab.com/mmdetection/v3.0/dab_detr/dab-detr_r50_8xb2-50e_coco/dab-detr_r50_8xb2-50e_coco_20221122_120837-c1035c8c.pth) \| [log](https://download.openmmlab.com/mmdetection/v3.0/dab_detr/dab-detr_r50_8xb2-50e_coco/dab-detr_r50_8xb2-50e_coco_20221122_120837.log.json) |

## Citation

Expand Down
12 changes: 6 additions & 6 deletions configs/deformable_detr/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,13 +14,13 @@ DETR has been recently proposed to eliminate the need for many hand-designed com

## Results and Models

| Backbone | Model | Lr schd | box AP | Config | Download |
| :------: | :---------------------------------: | :-----: | :----: | :---------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
| R-50 | Deformable DETR | 50e | 44.5 | [config](./deformable-detr_r50_16xb2-50e_coco.py) | [model](https://download.openmmlab.com/mmdetection/v2.0/deformable_detr/deformable_detr_r50_16x2_50e_coco/deformable_detr_r50_16x2_50e_coco_20210419_220030-a12b9512.pth) \| [log](https://download.openmmlab.com/mmdetection/v2.0/deformable_detr/deformable_detr_r50_16x2_50e_coco/deformable_detr_r50_16x2_50e_coco_20210419_220030-a12b9512.log.json) |
| R-50 | + iterative bounding box refinement | 50e | 46.1 | [config](./deformable-detr_refine_r50_16xb2-50e_coco.py) | [model](https://download.openmmlab.com/mmdetection/v2.0/deformable_detr/deformable_detr_refine_r50_16x2_50e_coco/deformable_detr_refine_r50_16x2_50e_coco_20210419_220503-5f5dff21.pth) \| [log](https://download.openmmlab.com/mmdetection/v2.0/deformable_detr/deformable_detr_refine_r50_16x2_50e_coco/deformable_detr_refine_r50_16x2_50e_coco_20210419_220503-5f5dff21.log.json) |
| R-50 | ++ two-stage Deformable DETR | 50e | 46.8 | [config](./deformable-detr_refine_twostage_r50_16xb2-50e_coco.py) | [model](https://download.openmmlab.com/mmdetection/v2.0/deformable_detr/deformable_detr_twostage_refine_r50_16x2_50e_coco/deformable_detr_twostage_refine_r50_16x2_50e_coco_20210419_220613-9d28ab72.pth) \| [log](https://download.openmmlab.com/mmdetection/v2.0/deformable_detr/deformable_detr_twostage_refine_r50_16x2_50e_coco/deformable_detr_twostage_refine_r50_16x2_50e_coco_20210419_220613-9d28ab72.log.json) |
| Backbone | Model | Lr schd | box AP | Config | Download |
| :------: | :---------------------------------: | :-----: | :----: | :---------------------------------------------------------------: | :------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
| R-50 | Deformable DETR | 50e | 44.3 | [config](./deformable-detr_r50_16xb2-50e_coco.py) | [model](https://download.openmmlab.com/mmdetection/v3.0/deformable_detr/deformable-detr_r50_16xb2-50e_coco/deformable-detr_r50_16xb2-50e_coco_20221029_210934-6bc7d21b.pth) \| [log](https://download.openmmlab.com/mmdetection/v3.0/deformable_detr/deformable-detr_r50_16xb2-50e_coco/deformable-detr_r50_16xb2-50e_coco_20221029_210934.log.json) |
| R-50 | + iterative bounding box refinement | 50e | 46.2 | [config](./deformable-detr-refine_r50_16xb2-50e_coco.py) | [model](https://download.openmmlab.com/mmdetection/v3.0/deformable_detr/deformable-detr-refine_r50_16xb2-50e_coco/deformable-detr-refine_r50_16xb2-50e_coco_20221022_225303-844e0f93.pth) \| [log](https://download.openmmlab.com/mmdetection/v3.0/deformable_detr/deformable-detr-refine_r50_16xb2-50e_coco/deformable-detr-refine_r50_16xb2-50e_coco_20221022_225303.log.json) |
| R-50 | ++ two-stage Deformable DETR | 50e | 47.0 | [config](./deformable-detr-refine-twostage_r50_16xb2-50e_coco.py) | [model](https://download.openmmlab.com/mmdetection/v3.0/deformable_detr/deformable-detr-refine-twostage_r50_16xb2-50e_coco/deformable-detr-refine-twostage_r50_16xb2-50e_coco_20221021_184714-acc8a5ff.pth) \| [log](https://download.openmmlab.com/mmdetection/v3.0/deformable_detr/deformable-detr-refine-twostage_r50_16xb2-50e_coco/deformable-detr-refine-twostage_r50_16xb2-50e_coco_20221021_184714.log.json) |

# NOTE
### NOTE

1. All models are trained with batch size 32.
2. The performance is unstable. `Deformable DETR` and `iterative bounding box refinement` may fluctuate about 0.3 mAP. `two-stage Deformable DETR` may fluctuate about 0.2 mAP.
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
_base_ = 'deformable-detr-refine_r50_16xb2-50e_coco.py'
model = dict(as_two_stage=True)

This file was deleted.

4 changes: 2 additions & 2 deletions configs/deformable_detr/metafile.yml
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ Models:

- Name: deformable-detr_refine_r50_16xb2-50e_coco
In Collection: Deformable DETR
Config: configs/deformable_detr/deformable-detr_refine_r50_16xb2-50e_coco.py
Config: configs/deformable_detr/deformable-detr-refine_r50_16xb2-50e_coco.py
Metadata:
Epochs: 50
Results:
Expand All @@ -45,7 +45,7 @@ Models:

- Name: deformable-detr_refine_twostage_r50_16xb2-50e_coco
In Collection: Deformable DETR
Config: configs/deformable_detr/deformable-detr_refine_twostage_r50_16xb2-50e_coco.py
Config: configs/deformable_detr/deformable-detr-refine-twostage_r50_16xb2-50e_coco.py
Metadata:
Epochs: 50
Results:
Expand Down
6 changes: 3 additions & 3 deletions configs/detr/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,9 @@ We present a new method that views object detection as a direct set prediction p

## Results and Models

| Backbone | Model | Lr schd | Mem (GB) | Inf time (fps) | box AP | Config | Download |
| :------: | :---: | :-----: | :------: | :------------: | :----: | :------------------------------------: | :----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
| R-50 | DETR | 150e | 7.9 | | 40.1 | [config](./detr_r50_8xb2-150e_coco.py) | [model](https://download.openmmlab.com/mmdetection/v2.0/detr/detr_r50_8x2_150e_coco/detr_r50_8x2_150e_coco_20201130_194835-2c4b8974.pth) \| [log](https://download.openmmlab.com/mmdetection/v2.0/detr/detr_r50_8x2_150e_coco/detr_r50_8x2_150e_coco_20201130_194835.log.json) |
| Backbone | Model | Lr schd | Mem (GB) | Inf time (fps) | box AP | Config | Download |
| :------: | :---: | :-----: | :------: | :------------: | :----: | :------------------------------------: | :--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
| R-50 | DETR | 150e | 7.9 | | 39.9 | [config](./detr_r50_8xb2-150e_coco.py) | [model](https://download.openmmlab.com/mmdetection/v3.0/detr/detr_r50_8xb2-150e_coco/detr_r50_8xb2-150e_coco_20221023_153551-436d03e8.pth) \| [log](https://download.openmmlab.com/mmdetection/v3.0/detr/detr_r50_8xb2-150e_coco/detr_r50_8xb2-150e_coco_20221023_153551.log.json) |

## Citation

Expand Down
10 changes: 7 additions & 3 deletions configs/dino/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,13 @@ We present DINO (DETR with Improved deNoising anchOr boxes), a state-of-the-art

## Results and Models

| Backbone | Model | Lr schd | box AP | Config | Download |
| :------: | :---: | :-----: | :----: | :------------------------------------------: | :------: |
| R-50 | DINO | 12e | 49.0 | [config](./dino_4scale_r50_8xb2-12e_coco.py) | |
| Backbone | Model | Lr schd | box AP | Config | Download |
| :------: | :---------: | :-----: | :----: | :------------------------------------------: | :--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
| R-50 | DINO-4scale | 12e | 49.0 | [config](./dino-4scale_r50_8xb2-12e_coco.py) | [model](https://download.openmmlab.com/mmdetection/v3.0/dino/dino-4scale_r50_8xb2-12e_coco/dino-4scale_r50_8xb2-12e_coco_20221202_182705-55b2bba2.pth) \| [log](https://download.openmmlab.com/mmdetection/v3.0/dino/dino-4scale_r50_8xb2-12e_coco/dino-4scale_r50_8xb2-12e_coco_20221202_182705.log.json) |

### NOTE

The performance is unstable. `DINO-4scale` with `R-50` may fluctuate about 0.4 mAP.

## Citation

Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
_base_ = './dino_4scale_r50_8xb2-12e_coco.py'
_base_ = './dino-4scale_r50_8xb2-12e_coco.py'
max_epochs = 24
train_cfg = dict(
type='EpochBasedTrainLoop', max_epochs=max_epochs, val_interval=1)
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
_base_ = './dino_4scale_r50_8xb2-12e_coco.py'
_base_ = './dino-4scale_r50_8xb2-12e_coco.py'
max_epochs = 36
train_cfg = dict(
type='EpochBasedTrainLoop', max_epochs=max_epochs, val_interval=1)
Expand Down
Loading

0 comments on commit 4c11117

Please sign in to comment.