Refine projects readme and model zoo (#131)

* refine box ap * refine model zoo * refine code format Co-authored-by: ntianhe ren <[email protected]>
IDEA-Research · Nov 15, 2022 · 5c0f5da · 5c0f5da
1 parent 0e3d42c
commit 5c0f5da
Show file tree

Hide file tree

Showing 12 changed files with 47 additions and 74 deletions.
diff --git a/demo/demo.py b/demo/demo.py
@@ -11,14 +11,14 @@
 import cv2
 import tqdm
 
-sys.path.insert(0, "./")  # noqa
-
 from demo.predictors import VisualizationDemo
 from detectron2.checkpoint import DetectionCheckpointer
 from detectron2.config import LazyConfig, instantiate
 from detectron2.data.detection_utils import read_image
 from detectron2.utils.logger import setup_logger
 
+sys.path.insert(0, "./")  # noqa
+
 
 # constants
 WINDOW_NAME = "COCO detections"

diff --git a/docs/source/tutorials/Model_Zoo.md b/docs/source/tutorials/Model_Zoo.md
@@ -24,14 +24,14 @@ Here we provides our pretrained baselines with **detrex**. And more pretrained w
 <td align="center">R50</td>
 <td align="center">IN1k</td>
 <td align="center">50</td>
-<td align="center">46.99</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.19</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>
@@ -52,7 +52,7 @@ Here we provides our pretrained baselines with **detrex**. And more pretrained w
 <td align="center">R-50</td>
 <td align="center">IN1k</td>
 <td align="center">50</td>
-<td align="center">41.60</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>
 </tbody></table>
@@ -72,35 +72,35 @@ Here we provides our pretrained baselines with **detrex**. And more pretrained w
 <td align="center">R50</td>
 <td align="center">IN1k</td>
 <td align="center">50</td>
-<td align="center">43.28</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_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">43.98</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_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.17</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.01</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.70</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>
@@ -121,7 +121,7 @@ Here we provides our pretrained baselines with **detrex**. And more pretrained w
 <td align="center">R50</td>
 <td align="center">IN1k</td>
 <td align="center">50</td>
-<td align="center">44.73</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>
 </tbody></table>
@@ -143,87 +143,87 @@ Here we provides our pretrained baselines with **detrex**. And more pretrained w
 <td align="center">IN1k</td>
 <td align="center">12</td>
 <td align="center">100</td>
-<td align="center">49.19</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_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.45</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.60</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">49.95</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>
  <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.30</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.50</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">52.96</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.83</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.92</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.93</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_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.09</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>
 </tbody></table>

diff --git a/projects/conditional_detr/README.md b/projects/conditional_detr/README.md
@@ -25,7 +25,7 @@ Here we provide the pretrained `Conditional-DETR` weights based on detrex.
 <td align="center">R-50</td>
 <td align="center">IN1k</td>
 <td align="center">50</td>
-<td align="center">41.60</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>
 </tbody></table>
@@ -46,15 +46,15 @@ Here we provide the pretrained `Conditional-DETR` weights based on detrex.
 <td align="center">R-50</td>
 <td align="center">IN1k</td>
 <td align="center">50</td>
-<td align="center">41.02</td>
+<td align="center">41.0</td>
 <td align="center"> <a href="https://github.com/IDEA-Research/detrex-storage/releases/download/v0.1.0/converted_conditional_detr_r50_50ep.pth">model</a></td>
 </tr>
 <!-- ROW: conditional_detr_r101_50ep -->
  <tr><td align="left"><a href="configs/conditional_detr_r101_50ep.py">Conditional-DETR-R101</a></td>
 <td align="center">R-101</td>
 <td align="center">IN1k</td>
 <td align="center">50</td>
-<td align="center">42.97</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>
 </tbody></table>

diff --git a/projects/dab_deformable_detr/README.md b/projects/dab_deformable_detr/README.md
@@ -25,15 +25,15 @@ Here we provide the pretrained `DAB-Deformable-DETR` weights based on detrex.
 <td align="center">R-50</td>
 <td align="center">IN1k</td>
 <td align="center">50</td>
-<td align="center">49.01</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>
 <!-- ROW: dab_deformable_detr_r50_two_stage_50ep -->
  <tr><td align="left"><a href="configs/dab_deformable_detr_r50_two_stage_50ep.py">DAB-Deformable-DETR-R50-Two-Stage</a></td>
 <td align="center">R-50</td>
 <td align="center">IN1k</td>
 <td align="center">50</td>
-<td align="center">49.70</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>

diff --git a/projects/dab_deformable_detr/configs/dab_deformable_detr_two_stage_900query_50ep.py b/projects/dab_deformable_detr/configs/dab_deformable_detr_two_stage_900query_50ep.py
diff --git a/projects/dab_deformable_detr/configs/models/dab_deformable_detr_r50.py b/projects/dab_deformable_detr/configs/models/dab_deformable_detr_r50.py
@@ -112,4 +112,3 @@
         aux_weight_dict.update({k + f"_{i}": v for k, v in base_weight_dict.items()})
     weight_dict.update(aux_weight_dict)
     model.criterion.weight_dict = weight_dict
-
diff --git a/projects/dab_detr/README.md b/projects/dab_detr/README.md
@@ -25,23 +25,23 @@ Here we provide the pretrained `DAB-DETR` weights based on detrex.
 <td align="center">R-50</td>
 <td align="center">IN1k</td>
 <td align="center">50</td>
-<td align="center">43.28</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>
 <!-- ROW: dab_detr_r101_50ep -->
  <tr><td align="left"><a href="configs/dab_detr_r101_50ep.py">DAB-DETR-R101</a></td>
 <td align="center">R-101</td>
 <td align="center">IN1k</td>
 <td align="center">50</td>
-<td align="center">43.98</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>
 <!-- ROW: dab_detr_swin_t_in1k_50ep -->
  <tr><td align="left"><a href="configs/dab_detr_swin_t_in1k_50ep.py">DAB-DETR-Swin-T</a></td>
 <td align="center">Swin-T</td>
 <td align="center">IN1k</td>
 <td align="center">50</td>
-<td align="center">45.17</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>
 </tbody></table>

diff --git a/projects/deformable_detr/README.md b/projects/deformable_detr/README.md
@@ -26,15 +26,15 @@ Here we provide the pretrained `Deformable-DETR` weights based on detrex.
 <td align="center">R-50</td>
 <td align="center">IN1k</td>
 <td align="center">50</td>
-<td align="center">46.99</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>
 <!-- ROW: deformable_detr_r50_two_stage_50ep -->
  <tr><td align="left"><a href="configs/deformable_detr_r50_two_stage_50ep.py">Deformable-DETR-R50 + Box-Refinement + Two-Stage</a></td>
 <td align="center">R-50</td>
 <td align="center">IN1k</td>
 <td align="center">50</td>
-<td align="center">48.19</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>

diff --git a/projects/detr/README.md b/projects/detr/README.md
@@ -28,15 +28,15 @@ Here we provides the weights which are converted by [converter.py](./converter.p
 <td align="center">R-50</td>
 <td align="center">IN1k</td>
 <td align="center">500</td>
-<td align="center">42.01</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>
 </tr>
 <!-- ROW: detr_r50 -->
  <tr><td align="left"><a>DETR-R101</a></td>
 <td align="center">R-50</td>
 <td align="center">IN1k</td>
 <td align="center">500</td>
-<td align="center">43.52</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>
 </tbody></table>
Original file line number	Diff line number	Diff line change
Expand Up		@@ -112,4 +112,3 @@
		aux_weight_dict.update({k + f"_{i}": v for k, v in base_weight_dict.items()})
		weight_dict.update(aux_weight_dict)
		model.criterion.weight_dict = weight_dict