Spaces:
Runtime error
Runtime error
| # Model Zoo | |
| ## Common Settings | |
| - All COCO models were trained on `coco_2017_train` and evaluated on `coco_2017_val`. | |
| - All models were trained using distributed training. | |
| - Most models were trained with 50 epochs settings (~51 COCO epochs) with multi-step LR scheduler which is the common setting in DETR-like methods. | |
| ## COCO Object Detection Baselines | |
| Here we provides our pretrained baselines with **detrex**. And more pretrained weights will be released in the future version. We also provide our converted pretrained weights for the users which will be marked as `(converted)`. | |
| ### DETR | |
| <table class="docutils"><tbody> | |
| <!-- START TABLE --> | |
| <!-- TABLE HEADER --> | |
| <th valign="bottom">Name</th> | |
| <th valign="bottom">Backbone</th> | |
| <th valign="bottom">Pretrained</th> | |
| <th valign="bottom">Epochs</th> | |
| <th valign="bottom">box<br/>AP</th> | |
| <th valign="bottom">Download</th> | |
| <!-- TABLE BODY --> | |
| <!-- ROW: detr_r50 --> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/detr/configs/detr_r50_300ep.py">DETR-R50 (converted)</a></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">500</td> | |
| <td align="center">42.0</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.0/converted_detr_r50_500ep.pth">model</a></td> | |
| <!-- ROW: detr_r50_dc5 --> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/detr/configs/detr_r50_dc5_300ep.py">DETR-R50-DC5 (converted)</a></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">500</td> | |
| <td align="center">43.4</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_detr_r50_dc5.pth">model</a></td> | |
| </tr> | |
| <!-- ROW: detr_r101 --> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/detr/configs/detr_r101_300ep.py">DETR-R101 (converted)</a></td> | |
| <td align="center">R-101</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">500</td> | |
| <td align="center">43.5</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.0/converted_detr_r101_500ep.pth">model</a></td> | |
| </tr> | |
| <!-- ROW: detr_r101_dc5 --> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/detr/configs/detr_r101_dc5_300ep.py">DETR-R101-DC5 (converted)</a></td> | |
| <td align="center">R-101</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">500</td> | |
| <td align="center">44.9</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_detr_r101_dc5.pth">model</a></td> | |
| </tr> | |
| </tbody></table> | |
| ### Deformable-DETR | |
| <table class="docutils"><tbody> | |
| <!-- START TABLE --> | |
| <!-- TABLE HEADER --> | |
| <th valign="bottom">Name</th> | |
| <th valign="bottom">Backbone</th> | |
| <th valign="bottom">Pretrained</th> | |
| <th valign="bottom">Epochs</th> | |
| <th valign="bottom">box<br/>AP</th> | |
| <th valign="bottom">Download</th> | |
| <!-- TABLE BODY --> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/deformable_detr/configs/deformable_detr_r50_with_box_refinement_50ep.py"> Deformable-DETR + Box Refinement </a> </td> | |
| <td align="center">R50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">47.0</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.1/deformable_detr_with_box_refinement_50ep_new.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/deformable_detr/configs/deformable_detr_r50_two_stage_50ep.py"> Deformable-DETR + Box Refinement + Two Stage </a> </td> | |
| <td align="center">R50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">48.2</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.1/deformable_detr_r50_two_stage_50ep_new.pth"> model </a></td> | |
| </tr> | |
| </tbody></table> | |
| ### Anchor-DETR | |
| <table class="docutils"><tbody> | |
| <!-- START TABLE --> | |
| <!-- TABLE HEADER --> | |
| <th valign="bottom">Name</th> | |
| <th valign="bottom">Backbone</th> | |
| <th valign="bottom">Pretrain</th> | |
| <th valign="bottom">Epochs</th> | |
| <th valign="bottom">box<br/>AP</th> | |
| <th valign="bottom">download</th> | |
| <!-- TABLE BODY --> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/anchor_detr/configs/anchor_detr_r50_50ep.py">Anchor-DETR-R50 </a></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">41.9</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/anchor_detr_r50_50ep.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/anchor_detr/configs/anchor_detr_r50_50ep.py">Anchor-DETR-R50 (converted)</a></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">42.2</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_anchor_detr_r50_50ep.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/anchor_detr/configs/detr_r50_dc5_300ep.py">Anchor-DETR-R50-DC5 (converted)</a></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">44.2</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_anchor_detr_r50_dc5_50ep.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/anchor_detr/configs/detr_r101_300ep.py">Anchor-DETR-R101 (converted)</a></td> | |
| <td align="center">R-101</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">43.5</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_anchor_detr_r101_50ep.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/anchor_detr/configs/detr_r101_dc5_300ep.py">Anchor-DETR-R101-DC5 (converted)</a></td> | |
| <td align="center">R-101</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">45.1</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_anchor_detr_r101_dc5_50ep.pth">model</a></td> | |
| </tr> | |
| </tbody></table> | |
| ### Conditional-DETR | |
| <table class="docutils"><tbody> | |
| <!-- START TABLE --> | |
| <!-- TABLE HEADER --> | |
| <th valign="bottom">Name</th> | |
| <th valign="bottom">Backbone</th> | |
| <th valign="bottom">Pretrain</th> | |
| <th valign="bottom">Epochs</th> | |
| <th valign="bottom">box<br/>AP</th> | |
| <th valign="bottom">download</th> | |
| <!-- TABLE BODY --> | |
| <!-- ROW: conditional_detr_r50_50ep --> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/conditional_detr/configs/conditional_detr_r50_50ep.py">Conditional-DETR-R50</a></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">41.6</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.0/conditional_detr_r50_50ep.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/conditional_detr/configs/conditional_detr_r50_dc5_50ep.py">Conditional-DETR-R50-DC5 (converted)</a></td> | |
| <td align="center">R-50-DC5</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">43.8</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_conditional_detr_r50_dc5.pth">model</a></td> | |
| </tr> | |
| <!-- ROW: conditional_detr_r101_50ep --> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/conditional_detr/configs/conditional_detr_r101_50ep.py">Conditional-DETR-R101 (converted)</a></td> | |
| <td align="center">R-101</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">43.0</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.0/converted_conditional_detr_r101_50ep.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/conditional_detr/configs/conditional_detr_r101_dc5_50ep.py">Conditional-DETR-R101-DC5 (converted)</a></td> | |
| <td align="center">R-101-DC5</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">45.1</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_conditional_detr_r101_dc5.pth">model</a></td> | |
| </tr> | |
| </tbody></table> | |
| ### DAB-DETR | |
| <table class="docutils"><tbody> | |
| <!-- START TABLE --> | |
| <!-- TABLE HEADER --> | |
| <th valign="bottom">Name</th> | |
| <th valign="bottom">Backbone</th> | |
| <th valign="bottom">Pretrained</th> | |
| <th valign="bottom">Epochs</th> | |
| <th valign="bottom">box<br/>AP</th> | |
| <th valign="bottom">Download</th> | |
| <!-- TABLE BODY --> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dab_detr/configs/dab_detr_r50_50ep.py"> DAB-DETR-R50 </a> </td> | |
| <td align="center">R50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">43.3</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.0/dab_detr_r50_50ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dab_detr/configs/dab_detr_r50_3patterns_50ep.py">DAB-DETR-R50-3patterns (converted)</a></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">42.8</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_dab_detr_r50_3patterns.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dab_detr/configs/dab_detr_r50_dc5_50ep.py">DAB-DETR-R50-DC5 (converted)</a></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">44.6</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_dab_detr_r50_dc5.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dab_detr/configs/dab_detr_r50_dc5_3patterns_50ep.py">DAB-DETR-R50-DC5-3patterns (converted)</a></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">45.7</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_dab_detr_r50_dc5_3patterns.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dab_detr/configs/dab_detr_r101_50ep.py"> DAB-DETR-R101 </a> </td> | |
| <td align="center">R101</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">44.0</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.0/dab_detr_r101_50ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dab_detr/configs/dab_detr_r50_dc5_3patterns_50ep.py">DAB-DETR-R101-DC5 (converted)</a></td> | |
| <td align="center">R-101</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">45.7</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_detr_r101_dc5.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dab_detr/configs/dab_detr_swin_t_in1k_50ep.py"> DAB-DETR-Swin-T </a> </td> | |
| <td align="center">Swin-Tiny-224</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">45.2</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.0/dab_detr_swin_t_in1k_50ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dab_deformable_detr/configs/dab_deformable_detr_r50_50ep.py"> DAB-Deformable-DETR-R50 </a> </td> | |
| <td align="center">R50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">49.0</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.0/dab_deformable_detr_r50_50ep_49AP.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dab_deformable_detr/configs/dab_deformable_detr_r50_two_stage_50ep.py"> DAB-Deformable-DETR-R50-Two-Stage </a> </td> | |
| <td align="center">R50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">49.7</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.0/dab_deformable_detr_r50_two_stage_49_7AP.pth"> model </a></td> | |
| </tr> | |
| </tbody></table> | |
| ### DN-DETR | |
| <table class="docutils"><tbody> | |
| <!-- START TABLE --> | |
| <!-- TABLE HEADER --> | |
| <th valign="bottom">Name</th> | |
| <th valign="bottom">Backbone</th> | |
| <th valign="bottom">Pretrained</th> | |
| <th valign="bottom">Epochs</th> | |
| <th valign="bottom">box<br/>AP</th> | |
| <th valign="bottom">Download</th> | |
| <!-- TABLE BODY --> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dn_detr/configs/dn_detr_r50_50ep.py"> DN-DETR-R50 </a> </td> | |
| <td align="center">R50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">44.7</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.0/dn_detr_r50_50ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dn_detr/configs/dn_detr_r50_dc5_50ep.py"> DN-DETR-R50-DC5 (converted) </a> </td> | |
| <td align="center">R50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">46.3</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_dn_detr_r50_dc5.pth"> model </a></td> | |
| </tr> | |
| </tbody></table> | |
| ### DINO | |
| **Pretrained DINO with ResNet Backbone** | |
| <table class="docutils"><tbody> | |
| <!-- START TABLE --> | |
| <!-- TABLE HEADER --> | |
| <th valign="bottom">Name</th> | |
| <th valign="bottom">Backbone</th> | |
| <th valign="bottom">Pretrained</th> | |
| <th valign="bottom">Epochs</th> | |
| <th valign="bottom">Denoising Queries</th> | |
| <th valign="bottom">box<br/>AP</th> | |
| <th valign="bottom">Download</th> | |
| <!-- TABLE BODY --> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_r50_4scale_12ep.py"> DINO-R50-4scale </a> </td> | |
| <td align="center">R50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">49.2</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.0/dino_r50_4scale_12ep_49_2AP.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_r50_4scale_12ep.py">DINO-R50-4scale (hacked trainer)</a></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">49.4</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.4.0/dino_r50_4scale_12ep_hacked_trainer.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left">DINO-R50-4scale <b> with EMA</b></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">49.4</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/dino_r50_4scale_12ep_with_ema.pth">model</a> </td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_r50_5scale_12ep.py"> DINO-R50-5scale </a> </td> | |
| <td align="center">R50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">49.6</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.1/dino_r50_5scale_12ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_r50_4scale_12ep_300dn.py"> DINO-R50-4scale </a> </td> | |
| <td align="center">R50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">300</td> | |
| <td align="center">49.5</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.0/dino_r50_4scale_12ep_300dn.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_r50_4scale_24ep.py"> DINO-R50-4scale </a> </td> | |
| <td align="center">R50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">24</td> | |
| <td align="center">100</td> | |
| <td align="center">50.6</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.1/dino_r50_4scale_24ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_r101_4scale_12ep.py"> DINO-R101-4scale </a> </td> | |
| <td align="center">R101</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">50.0</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.1/dino_r101_4scale_12ep.pth"> model </a></td> | |
| </tr> | |
| </tbody></table> | |
| **Pretrained DINO with Swin-Transformer Backbone** | |
| <table class="docutils"><tbody> | |
| <!-- START TABLE --> | |
| <!-- TABLE HEADER --> | |
| <th valign="bottom">Name</th> | |
| <th valign="bottom">Backbone</th> | |
| <th valign="bottom">Pretrained</th> | |
| <th valign="bottom">Epochs</th> | |
| <th valign="bottom">Denoising Queries</th> | |
| <th valign="bottom">box<br/>AP</th> | |
| <th valign="bottom">Download</th> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_swin_tiny_224_4scale_12ep.py"> DINO-Swin-T-224-4scale </a> </td> | |
| <td align="center">Swin-Tiny-224</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">51.3</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.1/dino_swin_tiny_224_4scale_12ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_swin_tiny_224_4scale_12ep.py"> DINO-Swin-T-224-4scale </a> </td> | |
| <td align="center">Swin-Tiny-224</td> | |
| <td align="center">IN22k to IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">52.5</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.1/dino_swin_tiny_224_22kto1k_finetune_4scale_12ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_swin_small_224_4scale_12ep.py"> DINO-Swin-S-224-4scale </a> </td> | |
| <td align="center">Swin-Small-224</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">53.0</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.1/dino_swin_small_224_4scale_12ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_swin_base_384_4scale_12ep.py"> DINO-Swin-B-384-4scale </a> </td> | |
| <td align="center">Swin-Base-384</td> | |
| <td align="center">IN22k to IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">55.8</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.1/dino_swin_base_384_4scale_12ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_swin_large_224_4scale_12ep.py"> DINO-Swin-L-224-4scale </a> </td> | |
| <td align="center">Swin-Large-224</td> | |
| <td align="center">IN22k to IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">56.9</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.1/dino_swin_large_224_4scale_12ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_swin_large_384_4scale_12ep.py"> DINO-Swin-L-384-4scale </a> </td> | |
| <td align="center">Swin-Large-384</td> | |
| <td align="center">IN22k to IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">56.9</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.1/dino_swin_large_4scale_12ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_swin_large_384_5scale_12ep.py"> DINO-Swin-L-384-5scale </a> </td> | |
| <td align="center">Swin-Large-384</td> | |
| <td align="center">IN22k to IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">57.5</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.1/dino_swin_large_384_5scale_12ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_swin_large_384_4scale_36ep.py"> DINO-Swin-L-384-4scale </a> </td> | |
| <td align="center">Swin-Large-384</td> | |
| <td align="center">IN22k to IN1k</td> | |
| <td align="center">36</td> | |
| <td align="center">100</td> | |
| <td align="center">58.1</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.0/dino_swin_large_384_4scale_36ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_swin_large_384_5scale_36ep.py"> DINO-Swin-L-384-5scale </a> </td> | |
| <td align="center">Swin-Large-384</td> | |
| <td align="center">IN22k to IN1k</td> | |
| <td align="center">36</td> | |
| <td align="center">100</td> | |
| <td align="center">58.5</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.1/dino_swin_large_384_5scale_36ep.pth"> model </a></td> | |
| </tbody></table> | |
| **Pretrained DINO with FocalNet Backbone** | |
| <table class="docutils"><tbody> | |
| <!-- START TABLE --> | |
| <!-- TABLE HEADER --> | |
| <th valign="bottom">Name</th> | |
| <th valign="bottom">Backbone</th> | |
| <th valign="bottom">Pretrained</th> | |
| <th valign="bottom">Epochs</th> | |
| <th valign="bottom">Denoising Queries</th> | |
| <th valign="bottom">box<br/>AP</th> | |
| <th valign="bottom">Download</th> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_focalnet_large_lrf_384_4scale_12ep.py"> DINO-FocalNet-Large-4scale </a> </td> | |
| <td align="center">FocalNet-384-LRF-3Level</td> | |
| <td align="center">IN22k</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">57.5</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.1/dino_focal_large_lrf_384_4scale_12ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_focalnet_large_lrf_384_fl4_4scale_12ep.py"> DINO-FocalNet-Large-4scale </a> </td> | |
| <td align="center">FocalNet-384-LRF-4Level</td> | |
| <td align="center">IN22k</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">58.0</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.1/dino_focal_large_lrf_384_fl4_4scale_12ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_focalnet_large_lrf_384_fl4_5scale_36ep.py"> DINO-FocalNet-Large-5scale </a> </td> | |
| <td align="center">FocalNet-384-LRF-4Level</td> | |
| <td align="center">IN22k</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">58.5</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.1/dino_focalnet_large_lrf_384_fl4_5scale_12ep.pth"> model </a></td> | |
| </tr> | |
| </tbody></table> | |
| **Pretrained DINO with ViTDet Backbone** | |
| <table class="docutils"><tbody> | |
| <!-- START TABLE --> | |
| <!-- TABLE HEADER --> | |
| <th valign="bottom">Name</th> | |
| <th valign="bottom">Backbone</th> | |
| <th valign="bottom">Pretrained</th> | |
| <th valign="bottom">Epochs</th> | |
| <th valign="bottom">Denoising Queries</th> | |
| <th valign="bottom">box<br/>AP</th> | |
| <th valign="bottom">Download</th> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_vitdet_base_4scale_12ep.py"> DINO-ViTDet-Base-4scale </a> </td> | |
| <td align="center">ViT</td> | |
| <td align="center">IN1k, MAE</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">50.2</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.1/dino_vitdet_4scale_12ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_vitdet_base_4scale_50ep.py"> DINO-ViTDet-Base-4scale </a> </td> | |
| <td align="center">ViT</td> | |
| <td align="center">IN1k, MAE</td> | |
| <td align="center">50</td> | |
| <td align="center">100</td> | |
| <td align="center">55.0</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.1/dino_vitdet_base_4scale_50ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_vitdet_large_4scale_12ep.py"> DINO-ViTDet-Large-4scale </a> </td> | |
| <td align="center">ViT</td> | |
| <td align="center">IN1k, MAE</td> | |
| <td align="center">12</td> | |
| <td align="center">100</td> | |
| <td align="center">52.9</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.1/dino_vitdet_large_4scale_12ep.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/dino/configs/dino_vitdet_large_4scale_50ep.py"> DINO-ViTDet-Large-4scale </a> </td> | |
| <td align="center">ViT</td> | |
| <td align="center">IN1k, MAE</td> | |
| <td align="center">50</td> | |
| <td align="center">100</td> | |
| <td align="center">57.5</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.1/dino_vitdet_large_4scale_50ep.pth"> model </a></td> | |
| </tr> | |
| </tbody></table> | |
| ### H-Deformable-DETR | |
| <table class="docutils"><tbody> | |
| <!-- START TABLE --> | |
| <!-- TABLE HEADER --> | |
| <th valign="bottom">Name</th> | |
| <th valign="bottom">Backbone</th> | |
| <th valign="bottom">Pretrained</th> | |
| <th valign="bottom">Query</th> | |
| <th valign="bottom">Epochs</th> | |
| <th valign="bottom">box<br/>AP</th> | |
| <th valign="bottom">Download</th> | |
| <!-- TABLE BODY --> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/h_deformable_detr/configs/h_deformable_detr_r50_two_stage_12ep.py"> H-Deformable-DETR-R50 + tricks (detrex) </a> </td> | |
| <td align="center">R50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">300</td> | |
| <td align="center">12</td> | |
| <td align="center">49.1</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.4.0/h_deformable_detr_r50_two_stage_12ep_modified_train_net.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/h_deformable_detr/configs/h_deformable_detr_r50_two_stage_12ep.py"> H-Deformable-DETR-R50 + tricks (converted) </a> </td> | |
| <td align="center">R50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">300</td> | |
| <td align="center">12</td> | |
| <td align="center">48.9</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.0/r50_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_12eps.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/h_deformable_detr/configs/h_deformable_detr_r50_two_stage_36ep.py"> H-Deformable-DETR-R50 + tricks (converted) </a> </td> | |
| <td align="center">R50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">300</td> | |
| <td align="center">36</td> | |
| <td align="center">50.3</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.0/r50_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_36eps.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/h_deformable_detr/configs/h_deformable_detr_swin_tiny_two_stage_12ep.py"> H-Deformable-DETR-Swin-T + tricks (converted) </a> </td> | |
| <td align="center">Swin-Tiny</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">300</td> | |
| <td align="center">12</td> | |
| <td align="center">50.6</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.0/swin_tiny_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_12eps.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/h_deformable_detr/configs/h_deformable_detr_swin_tiny_two_stage_36ep.py"> H-Deformable-DETR-Swin-T + tricks (converted) </a> </td> | |
| <td align="center">Swin-Tiny</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">300</td> | |
| <td align="center">36</td> | |
| <td align="center">53.5</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.0/swin_tiny_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_36eps.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/h_deformable_detr/configs/h_deformable_detr_swin_large_two_stage_12ep.py"> H-Deformable-DETR-Swin-L + tricks (converted) </a> </td> | |
| <td align="center">Swin-Large</td> | |
| <td align="center">IN22k</td> | |
| <td align="center">300</td> | |
| <td align="center">12</td> | |
| <td align="center">56.2</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.0/swin_large_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_12eps.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/h_deformable_detr/configs/h_deformable_detr_swin_large_two_stage_36ep.py"> H-Deformable-DETR-Swin-L + tricks (converted) </a> </td> | |
| <td align="center">Swin-Large</td> | |
| <td align="center">IN22k</td> | |
| <td align="center">300</td> | |
| <td align="center">36</td> | |
| <td align="center">57.5</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.0/drop_path0.5_swin_large_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_36eps.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/h_deformable_detr/configs/h_deformable_detr_swin_large_two_stage_12ep_900queries.py"> H-Deformable-DETR-Swin-L + tricks (converted) </a> </td> | |
| <td align="center">Swin-Large</td> | |
| <td align="center">IN22k</td> | |
| <td align="center">900</td> | |
| <td align="center">12</td> | |
| <td align="center">56.4</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.0/swin_large_hybrid_branch_lambda1_group6_t1500_n900_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_12eps.pth"> model </a></td> | |
| </tr> | |
| <tr><td align="left"> <a href="https://github.com/IDEA-Research/detrex/blob/main/projects/h_deformable_detr/configs/h_deformable_detr_swin_large_two_stage_36ep_900queries.py"> H-Deformable-DETR-Swin-L + tricks (converted) </a> </td> | |
| <td align="center">Swin-Large</td> | |
| <td align="center">IN22k</td> | |
| <td align="center">300</td> | |
| <td align="center">36</td> | |
| <td align="center">57.5</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.2.0/drop_path0.5_swin_large_hybrid_branch_lambda1_group6_t1500_n900_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_36eps.pth"> model </a></td> | |
| </tr> | |
| </tbody></table> | |
| ### DETA | |
| <table class="docutils"><tbody> | |
| <!-- START TABLE --> | |
| <!-- TABLE HEADER --> | |
| <th valign="bottom">Name</th> | |
| <th valign="bottom">Backbone</th> | |
| <th valign="bottom">Pretrained</th> | |
| <th valign="bottom">Epochs</th> | |
| <th valign="bottom">box<br/>AP</th> | |
| <th valign="bottom">Download</th> | |
| <!-- TABLE BODY --> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/deta/configs/improved_deformable_detr_baseline_50ep.py">Improved-Deformable-DETR-R50 (converted)</a></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">50</td> | |
| <td align="center">49.8</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_deta_improved_deformable_baseline.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/deta/configs/deta_r50_5scale_12ep_bs8.py">DETA-R50-5scale (bs=8, 180000 iterations)</a></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">50.0</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.4.0/deta_r50_5scale_12ep_bs8.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/deta/configs/deta_r50_5scale_12ep.py">DETA-R50-5scale (with hacked train engine)</a></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">49.9</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.4.0/deta_r50_5scale_12ep_hacked_trainer.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/deta/configs/deta_r50_5scale_no_frozen_backbone.py">DETA-R50-5scale-12ep (no frozen backbone)</a></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">50.2</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.4.0/deta_r50_5scale_12ep_no_freeze_backbone.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/deta/configs/deta_r50_5scale_12ep.py">DETA-R50-5scale (converted)</a></td> | |
| <td align="center">R-50</td> | |
| <td align="center">IN1k</td> | |
| <td align="center">12</td> | |
| <td align="center">50.1</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_deta_r50_5scale_12ep.pth">model</a></td> | |
| </tr> | |
| <tr><td align="left"><a href="https://github.com/IDEA-Research/detrex/blob/main/projects/deta/configs/deta_r50_5scale_12ep.py">DETA-Swin-Large-finetune (converted)</a></td> | |
| <td align="center">Swin-Large-384</td> | |
| <td align="center">Object 365</td> | |
| <td align="center">24</td> | |
| <td align="center">62.9</td> | |
| <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.3.0/converted_deta_swin_o365_finetune.pth">model</a></td> | |
| </tr> | |
| </tbody></table> | |