Merge pull request #41 from aim-uofa/fcos_head_bn

Fcos head bn
aim-uofa · Apr 23, 2020 · f41b5f4 · f41b5f4
2 parents d0df12f + 4eb95f4
commit f41b5f4
Show file tree

Hide file tree

Showing 6 changed files with 89 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -35,9 +35,17 @@ Name | inf. time | box AP | download
 
 Name | inf. time | box AP | download
 --- |:---:|:---:|:---:
-[FCOS_RT_DLA_34_4x_shtw](configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_shared_towers.yaml) | 52 FPS | 39.1 | [model](https://cloudstor.aarnet.edu.au/plus/s/4vc3XwQezyhNvnB/download)
-[FCOS_RT_DLA_34_4x](configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn.yaml) | 46 FPS | 40.3 | [model](https://cloudstor.aarnet.edu.au/plus/s/zNPNyTkizaOOsUQ/download)
-[FCOS_RT_R_50_4x](configs/FCOS-Detection/FCOS_RT/MS_R_50_4x_syncbn.yaml) | 38 FPS | 40.2 | [model](https://cloudstor.aarnet.edu.au/plus/s/TlnlXUr6lNNSyoZ/download)
+[FCOS_RT_MS_DLA_34_4x_shtw](configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_shared_towers.yaml) | 52 FPS | 39.1 | [model](https://cloudstor.aarnet.edu.au/plus/s/4vc3XwQezyhNvnB/download)
+[FCOS_RT_MS_DLA_34_4x](configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn.yaml) | 46 FPS | 40.3 | [model](https://cloudstor.aarnet.edu.au/plus/s/zNPNyTkizaOOsUQ/download)
+[FCOS_RT_MS_R_50_4x](configs/FCOS-Detection/FCOS_RT/MS_R_50_4x_syncbn.yaml) | 38 FPS | 40.2 | [model](https://cloudstor.aarnet.edu.au/plus/s/TlnlXUr6lNNSyoZ/download)
+
+If you prefer BN in FCOS heads, please try the following models.
+
+Name | inf. time | box AP | download
+--- |:---:|:---:|:---:
+[FCOS_RT_MS_DLA_34_4x_shtw_bn](configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_shared_towers_bn_head.yaml) | 52 FPS | 38.9 | [model](https://cloudstor.aarnet.edu.au/plus/s/rdmHHSs4oCg7l7U/download)
+[FCOS_RT_MS_DLA_34_4x_bn](configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_bn_head.yaml) | 48 FPS | 39.4 | [model](https://cloudstor.aarnet.edu.au/plus/s/T5httPVo1VndbD4/download)
+[FCOS_RT_MS_R_50_4x_bn](configs/FCOS-Detection/FCOS_RT/MS_R_50_4x_syncbn_bn_head.yaml) | 40 FPS | 39.3 | [model](https://cloudstor.aarnet.edu.au/plus/s/dHNUNs0YxVhZAmg/download)
 
 *Inference time is measured on a NVIDIA 1080Ti with batch size 1. Real-time models use shorter side 512 for inference.*
 

diff --git a/configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_bn_head.yaml b/configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_bn_head.yaml
@@ -0,0 +1,26 @@
+_BASE_: "../Base-FCOS.yaml"
+INPUT:
+  MIN_SIZE_TRAIN: (256, 288, 320, 352, 384, 416, 448, 480, 512, 544, 576, 608)
+  MAX_SIZE_TRAIN: 900
+  MAX_SIZE_TEST: 736
+  MIN_SIZE_TEST: 512
+MODEL:
+  BACKBONE:
+    NAME: "build_fcos_dla_fpn_backbone"
+    FREEZE_AT: -1
+  WEIGHTS: "http://dl.yf.io/dla/models/imagenet/dla34-ba72cf86.pth"
+  DLA:
+    CONV_BODY: "DLA34"
+    NORM: "SyncBN"
+  FPN:
+    IN_FEATURES: ["level3", "level4", "level5"]
+  FCOS:
+    TOP_LEVELS: 0
+    SIZES_OF_INTEREST: [64, 128]
+    FPN_STRIDES: [8, 16, 32]
+    IN_FEATURES: ['p3', 'p4', 'p5']
+    NORM: "SyncBN"
+SOLVER:
+  STEPS: (300000, 340000)
+  MAX_ITER: 360000
+OUTPUT_DIR: "output/fcos/FCOS_RT_MS_DLA_34_4x_syncbn_bn_head"
diff --git a/configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_shared_towers.yaml b/configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_shared_towers.yaml
@@ -25,4 +25,4 @@ MODEL:
 SOLVER:
   STEPS: (300000, 340000)
   MAX_ITER: 360000
-OUTPUT_DIR: "output/fcos/MS_DLA_34_4x_syncbn_shared_towers"
+OUTPUT_DIR: "output/fcos/FCOS_RT_MS_DLA_34_4x_syncbn_shared_towers"
diff --git a/configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_shared_towers_bn_head.yaml b/configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_shared_towers_bn_head.yaml
@@ -0,0 +1,29 @@
+_BASE_: "../Base-FCOS.yaml"
+INPUT:
+  MIN_SIZE_TRAIN: (256, 288, 320, 352, 384, 416, 448, 480, 512, 544, 576, 608)
+  MAX_SIZE_TRAIN: 900
+  MAX_SIZE_TEST: 736
+  MIN_SIZE_TEST: 512
+MODEL:
+  BACKBONE:
+    NAME: "build_fcos_dla_fpn_backbone"
+    FREEZE_AT: -1
+  WEIGHTS: "http://dl.yf.io/dla/models/imagenet/dla34-ba72cf86.pth"
+  DLA:
+    CONV_BODY: "DLA34"
+    NORM: "SyncBN"
+  FPN:
+    IN_FEATURES: ["level3", "level4", "level5"]
+  FCOS:
+    TOP_LEVELS: 0
+    SIZES_OF_INTEREST: [64, 128]
+    FPN_STRIDES: [8, 16, 32]
+    IN_FEATURES: ['p3', 'p4', 'p5']
+    NUM_SHARE_CONVS: 4
+    NUM_BOX_CONVS: 0
+    NUM_CLS_CONVS: 0
+    NORM: "SyncBN"
+SOLVER:
+  STEPS: (300000, 340000)
+  MAX_ITER: 360000
+OUTPUT_DIR: "output/fcos/FCOS_RT_MS_DLA_34_4x_syncbn_shared_towers_bn_head"
diff --git a/configs/FCOS-Detection/FCOS_RT/MS_R_50_4x_syncbn.yaml b/configs/FCOS-Detection/FCOS_RT/MS_R_50_4x_syncbn.yaml
@@ -17,4 +17,4 @@ MODEL:
 SOLVER:
   STEPS: (300000, 340000)
   MAX_ITER: 360000
-OUTPUT_DIR: "output/fcos/FCOS_RT_MS_R_50_4x"
+OUTPUT_DIR: "output/fcos/FCOS_RT_MS_R_50_4x_syncbn"
diff --git a/configs/FCOS-Detection/FCOS_RT/MS_R_50_4x_syncbn_bn_head.yaml b/configs/FCOS-Detection/FCOS_RT/MS_R_50_4x_syncbn_bn_head.yaml
@@ -0,0 +1,21 @@
+_BASE_: "../Base-FCOS.yaml"
+INPUT:
+  MIN_SIZE_TRAIN: (256, 288, 320, 352, 384, 416, 448, 480, 512, 544, 576, 608)
+  MAX_SIZE_TRAIN: 900
+  MAX_SIZE_TEST: 736
+  MIN_SIZE_TEST: 512
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  RESNETS:
+    DEPTH: 50
+    NORM: "SyncBN"
+  FCOS:
+    TOP_LEVELS: 0
+    SIZES_OF_INTEREST: [64, 128]
+    FPN_STRIDES: [8, 16, 32]
+    IN_FEATURES: ['p3', 'p4', 'p5']
+    NORM: "SyncBN"
+SOLVER:
+  STEPS: (300000, 340000)
+  MAX_ITER: 360000
+OUTPUT_DIR: "output/fcos/FCOS_RT_MS_R_50_4x_syncbn_bn_head"