From d22575cc9472b842f1c21bb7775cdc084255c68a Mon Sep 17 00:00:00 2001
From: Jiaqi Wang <1155098160@link.cuhk.edu.hk>
Date: Fri, 31 May 2019 15:29:41 +0800
Subject: [PATCH] update Guided Anchoring benchmarks (#740)
---
configs/guided_anchoring/README.md | 38 +++++++++----------
.../models/anchor_heads/guided_anchor_head.py | 5 ++-
2 files changed, 23 insertions(+), 20 deletions(-)
diff --git a/configs/guided_anchoring/README.md b/configs/guided_anchoring/README.md
index 1d8bb00..3821944 100644
--- a/configs/guided_anchoring/README.md
+++ b/configs/guided_anchoring/README.md
@@ -17,25 +17,25 @@ We provide config files to reproduce the results in the CVPR 2019 paper for [Reg
The results on COCO 2017 val is shown in the below table. (results on test-dev are usually slightly higher than val).
-| Method | Backbone | Style | Lr schd | Mem (GB) | Train time (s/iter) | Inf time (fps) | AR 1000 | Download |
-| :----: | :-------------: | :-----: | :-----: | :------: | :-----------------: | :------------: | :-----: | :-------------------------------------------------------------------------------------------------------------------------------------------: |
-| GA-RPN | R-50-FPN | caffe | 1x | 5.0 | 0.55 | 13.3 | 68.5 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/guided_anchoring/ga_rpn_r50_caffe_fpn_1x_20190513-95e91886.pth) |
-| GA-RPN | R-101-FPN | caffe | 1x | - | - | - | 69.6 | - |
-| GA-RPN | X-101-32x4d-FPN | pytorch | 1x | - | - | - | 70.0 | - |
-| GA-RPN | X-101-64x4d-FPN | pytorch | 1x | - | - | - | 70.5 | - |
-
-
-| Method | Backbone | Style | Lr schd | Mem (GB) | Train time (s/iter) | Inf time (fps) | box AP | Download |
-| :------------: | :-------------: | :-----: | :-----: | :------: | :-----------------: | :------------: | :----: | :-------------------------------------------------------------------------------------------------------------------------------------------------: |
-| GA-Fast RCNN | R-50-FPN | caffe | 1x | 3.3 | 0.23 | 14.9 | 39.5 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/guided_anchoring/ga_fast_r50_caffe_fpn_1x_20190513-c5af9f8b.pth) |
-| GA-Faster RCNN | R-50-FPN | caffe | 1x | 5.1 | 0.64 | 9.6 | 39.9 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/guided_anchoring/ga_faster_r50_caffe_fpn_1x_20190513-a52b31fa.pth) |
-| GA-Faster RCNN | R-101-FPN | caffe | 1x | - | - | - | 41.5 | - |
-| GA-Faster RCNN | X-101-32x4d-FPN | pytorch | 1x | - | - | - | 42.9 | - |
-| GA-Faster RCNN | X-101-64x4d-FPN | pytorch | 1x | - | - | - | 43.9 | - |
-| GA-RetinaNet | R-50-FPN | caffe | 1x | 3.2 | 0.50 | 10.7 | 37.0 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/guided_anchoring/ga_retinanet_r50_caffe_fpn_1x_20190513-29905101.pth) |
-| GA-RetinaNet | R-101-FPN | caffe | 1x | - | - | - | 38.9 | - |
-| GA-RetinaNet | X-101-32x4d-FPN | pytorch | 1x | - | - | - | 40.3 | - |
-| GA-RetinaNet | X-101-64x4d-FPN | pytorch | 1x | - | - | - | 40.8 | - |
+| Method | Backbone | Style | Lr schd | Mem (GB) | Train time (s/iter) | Inf time (fps) | AR 1000 | Download |
+| :----: | :-------------: | :-----: | :-----: | :------: | :-----------------: | :------------: | :-----: | :--------------------------------------------------------------------------------------------------------------------------------------------: |
+| GA-RPN | R-50-FPN | caffe | 1x | 5.0 | 0.55 | 13.3 | 68.5 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/guided_anchoring/ga_rpn_r50_caffe_fpn_1x_20190513-95e91886.pth) |
+| GA-RPN | R-101-FPN | caffe | 1x | 7.1 | 0.66 | 9.8 | 69.6 | [model](https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmdetection/models/guided_anchoring/ga_rpn_r101_caffe_fpn_1x_20190523-91e0b817.pth) |
+| GA-RPN | X-101-32x4d-FPN | pytorch | 1x | 8.5 | 0.88 | 8.5 | 70.0 | [model](https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmdetection/models/guided_anchoring/ga_rpn_x101_32x4d_fpn_1x_20190523-a60df28c.pth) |
+| GA-RPN | X-101-64x4d-FPN | pytorch | 1x | 11.4 | 1.24 | 6.5 | 70.5 | [model](https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmdetection/models/guided_anchoring/ga_rpn_x101_64x4d_fpn_1x_20190523-9f2449ba.pth) |
+
+
+| Method | Backbone | Style | Lr schd | Mem (GB) | Train time (s/iter) | Inf time (fps) | box AP | Download |
+| :------------: | :-------------: | :-----: | :-----: | :------: | :-----------------: | :------------: | :----: | :--------------------------------------------------------------------------------------------------------------------------------------------------: |
+| GA-Fast RCNN | R-50-FPN | caffe | 1x | 3.3 | 0.23 | 14.9 | 39.5 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/guided_anchoring/ga_fast_r50_caffe_fpn_1x_20190513-c5af9f8b.pth) |
+| GA-Faster RCNN | R-50-FPN | caffe | 1x | 5.1 | 0.64 | 9.6 | 39.9 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/guided_anchoring/ga_faster_r50_caffe_fpn_1x_20190513-a52b31fa.pth) |
+| GA-Faster RCNN | R-101-FPN | caffe | 1x | 7.3 | 0.75 | 8.0 | 41.5 | [model](https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmdetection/models/guided_anchoring/ga_faster_r101_caffe_fpn_1x_20190523-9a711ec5.pth) |
+| GA-Faster RCNN | X-101-32x4d-FPN | pytorch | 1x | 8.7 | 0.97 | 7.1 | 42.9 | [model](https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmdetection/models/guided_anchoring/ga_faster_x101_32x4d_fpn_1x_20190523-8dc3e59a.pth) |
+| GA-Faster RCNN | X-101-64x4d-FPN | pytorch | 1x | 11.6 | 1.33 | 5.7 | 43.9 | [model](https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmdetection/models/guided_anchoring/ga_faster_x101_64x4d_fpn_1x_20190523-a8589c97.pth) |
+| GA-RetinaNet | R-50-FPN | caffe | 1x | 3.2 | 0.50 | 10.7 | 37.0 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/guided_anchoring/ga_retinanet_r50_caffe_fpn_1x_20190513-29905101.pth) |
+| GA-RetinaNet | R-101-FPN | caffe | 1x | 5.3 | 0.63 | 8.5 | 38.9 | [model](https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmdetection/models/guided_anchoring/ga_retinanet_r101_caffe_fpn_1x_20190523-792ad63d.pth) |
+| GA-RetinaNet | X-101-32x4d-FPN | pytorch | 1x | 6.7 | 0.87 | 7.5 | 40.3 | [model](https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmdetection/models/guided_anchoring/ga_retinanet_x101_32x4d_fpn_1x_20190523-4ec3f13c.pth) |
+| GA-RetinaNet | X-101-64x4d-FPN | pytorch | 1x | 9.6 | 1.22 | 5.8 | 40.8 | [model](https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmdetection/models/guided_anchoring/ga_retinanet_x101_64x4d_fpn_1x_20190523-013d1913.pth) |
diff --git a/mmdet/models/anchor_heads/guided_anchor_head.py b/mmdet/models/anchor_heads/guided_anchor_head.py
index da43aa8..0053cf5 100644
--- a/mmdet/models/anchor_heads/guided_anchor_head.py
+++ b/mmdet/models/anchor_heads/guided_anchor_head.py
@@ -114,7 +114,10 @@ class GuidedAnchorHead(AnchorHead):
gamma=2.0,
alpha=0.25,
loss_weight=1.0),
- loss_shape=dict(type='IoULoss', beta=0.2, loss_weight=1.0),
+ loss_shape=dict(type='IoULoss',
+ style='bounded',
+ beta=0.2,
+ loss_weight=1.0),
loss_cls=dict(type='CrossEntropyLoss',
use_sigmoid=True,
loss_weight=1.0),
--
GitLab