From 690a04b8e4d7f2c0efc70536928aaa444bea7385 Mon Sep 17 00:00:00 2001 From: Hao Chen Date: Fri, 24 Apr 2020 12:21:24 +0930 Subject: [PATCH] add RT_R_50 config --- README.md | 1 + configs/BlendMask/RT_R_50_4x.yaml | 17 +++++++++++++++++ 2 files changed, 18 insertions(+) create mode 100644 configs/BlendMask/RT_R_50_4x.yaml diff --git a/README.md b/README.md index 74010b0ae..13edb574d 100644 --- a/README.md +++ b/README.md @@ -47,6 +47,7 @@ Model | Name |inf. time | box AP | mask AP | download --- |:---:|:---:|:---:|:---:|:---: Mask R-CNN | [550_R_50_3x](configs/RCNN/550_R_50_FPN_3x.yaml) | 16FPS | 39.1 | 35.3 | BlendMask | [550_R_50_3x](configs/BlendMask/550_R_50_3x.yaml) | 28FPS | 38.7 | 34.5 | [model](https://cloudstor.aarnet.edu.au/plus/s/R3Qintf7N8UCiIt/download) +BlendMask | [RT_R_50_4x](configs/BlendMask/RT_R_50_4x.yaml) | 30FPS | 40.1 | 34.6 | [model](https://cloudstor.aarnet.edu.au/plus/s/fmmciLkyaOoY1Tc/download) BlendMask | [DLA_34_4x](configs/BlendMask/DLA_34_syncbn_4x.yaml) | 32FPS | 40.9 | 35.2 | [model](https://cloudstor.aarnet.edu.au/plus/s/Lx94rWNnZ8TRd2Y/download) Mask R-CNN | [R_50_1x](https://github.com/facebookresearch/detectron2/blob/master/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml) | 13FPS | 38.6 | 35.2 | BlendMask | [R_50_1x](configs/BlendMask/R_50_1x.yaml) | 14FPS | 39.9 | 35.8 | [model](https://cloudstor.aarnet.edu.au/plus/s/zoxXPnr6Hw3OJgK/download) diff --git a/configs/BlendMask/RT_R_50_4x.yaml b/configs/BlendMask/RT_R_50_4x.yaml new file mode 100644 index 000000000..6d00ecc82 --- /dev/null +++ b/configs/BlendMask/RT_R_50_4x.yaml @@ -0,0 +1,17 @@ +_BASE_: "Base-550.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" + BACKBONE: + FREEZE_AT: -1 +SOLVER: + STEPS: (300000, 340000) + MAX_ITER: 360000 +OUTPUT_DIR: "output/blendmask/RT_R_50_4x"