Skip to content

GitLab

"examples/vscode:/vscode.git/clone" did not exist on "4974b84564d25bd4b5c594db4e04cb885cc0a9ed"
Commit d22575cc authored by Jiaqi Wang's avatar Jiaqi Wang Committed by Kai Chen
Browse files

update Guided Anchoring benchmarks (#740)

parent a6857292
Hide whitespace changes
Inline Side-by-side
Showing with 23 additions and 20 deletions
configs/guided_anchoring/README.md
View file @ d22575cc
......@@ -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) |
......
mmdet/models/anchor_heads/guided_anchor_head.py
View file @ d22575cc
......@@ -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),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment