GHM

Last updated on Feb 23, 2021

GHM (R-101-FPN, 1x, pytorch)

Memory (M) 6000.0
inference time (s/im) 0.22727
File Size 217.80 MB
Training Data MS COCO
Training Resources 8x NVIDIA V100 GPUs
Training Time

Architecture GHM-C, ResNet, FPN, GHM-R
lr sched 1x
Memory (M) 6000.0
Backbone Layers 101
inference time (s/im) 0.22727
SHOW MORE
SHOW LESS
GHM (R-50-FPN, 1x, pytorch)

Memory (M) 4000.0
inference time (s/im) 0.30303
File Size 145.10 MB
Training Data MS COCO
Training Resources 8x NVIDIA V100 GPUs
Training Time

Architecture GHM-C, ResNet, FPN, GHM-R
lr sched 1x
Memory (M) 4000.0
Backbone Layers 50
inference time (s/im) 0.30303
SHOW MORE
SHOW LESS
GHM (X-101-32x4d-FPN, 1x, pytorch)

Memory (M) 7200.0
inference time (s/im) 0.19608
File Size 216.51 MB
Training Data MS COCO
Training Resources 8x NVIDIA V100 GPUs
Training Time

Architecture ResNeXt, GHM-C, FPN, GHM-R
lr sched 1x
Memory (M) 7200.0
Backbone Layers 101
inference time (s/im) 0.19608
SHOW MORE
SHOW LESS
GHM (X-101-64x4d-FPN, 1x, pytorch)

Memory (M) 10300.0
inference time (s/im) 0.19231
File Size 366.59 MB
Training Data MS COCO
Training Resources 8x NVIDIA V100 GPUs
Training Time

Architecture ResNeXt, GHM-C, FPN, GHM-R
lr sched 1x
Memory (M) 10300.0
Backbone Layers 101
inference time (s/im) 0.19231
SHOW MORE
SHOW LESS
README.md

Gradient Harmonized Single-stage Detector

Introduction

[ALGORITHM]

@inproceedings{li2019gradient,
  title={Gradient Harmonized Single-stage Detector},
  author={Li, Buyu and Liu, Yu and Wang, Xiaogang},
  booktitle={AAAI Conference on Artificial Intelligence},
  year={2019}
}

Results and Models

Backbone Style Lr schd Mem (GB) Inf time (fps) box AP Config Download
R-50-FPN pytorch 1x 4.0 3.3 37.0 config model | log
R-101-FPN pytorch 1x 6.0 4.4 39.1 config model | log
X-101-32x4d-FPN pytorch 1x 7.2 5.1 40.7 config model | log
X-101-64x4d-FPN pytorch 1x 10.3 5.2 41.4 config model | log

Results

Object Detection on COCO minival

Object Detection on COCO minival
MODEL BOX AP
GHM (X-101-64x4d-FPN, 1x, pytorch) 41.4
GHM (X-101-32x4d-FPN, 1x, pytorch) 40.7
GHM (R-101-FPN, 1x, pytorch) 39.1
GHM (R-50-FPN, 1x, pytorch) 37.0