forked from open-mmlab/mmdetection
-
Notifications
You must be signed in to change notification settings - Fork 0
/
metafile.yml
101 lines (97 loc) · 3.03 KB
/
metafile.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
Collections:
- Name: GHM
Metadata:
Training Data: COCO
Training Techniques:
- SGD with Momentum
- Weight Decay
Training Resources: 8x V100 GPUs
Architecture:
- GHM-C
- GHM-R
- FPN
- ResNet
Paper:
URL: https://arxiv.org/abs/1811.05181
Title: 'Gradient Harmonized Single-stage Detector'
README: configs/ghm/README.md
Code:
URL: https://github.com/open-mmlab/mmdetection/blob/v2.0.0/mmdet/models/losses/ghm_loss.py#L21
Version: v2.0.0
Models:
- Name: retinanet_ghm_r50_fpn_1x_coco
In Collection: GHM
Config: configs/ghm/retinanet_ghm_r50_fpn_1x_coco.py
Metadata:
Training Memory (GB): 4.0
inference time (ms/im):
- value: 303.03
hardware: V100
backend: PyTorch
batch size: 1
mode: FP32
resolution: (800, 1333)
Epochs: 12
Results:
- Task: Object Detection
Dataset: COCO
Metrics:
box AP: 37.0
Weights: https://download.openmmlab.com/mmdetection/v2.0/ghm/retinanet_ghm_r50_fpn_1x_coco/retinanet_ghm_r50_fpn_1x_coco_20200130-a437fda3.pth
- Name: retinanet_ghm_r101_fpn_1x_coco
In Collection: GHM
Config: configs/ghm/retinanet_ghm_r101_fpn_1x_coco.py
Metadata:
Training Memory (GB): 6.0
inference time (ms/im):
- value: 227.27
hardware: V100
backend: PyTorch
batch size: 1
mode: FP32
resolution: (800, 1333)
Epochs: 12
Results:
- Task: Object Detection
Dataset: COCO
Metrics:
box AP: 39.1
Weights: https://download.openmmlab.com/mmdetection/v2.0/ghm/retinanet_ghm_r101_fpn_1x_coco/retinanet_ghm_r101_fpn_1x_coco_20200130-c148ee8f.pth
- Name: retinanet_ghm_x101_32x4d_fpn_1x_coco
In Collection: GHM
Config: configs/ghm/retinanet_ghm_x101_32x4d_fpn_1x_coco.py
Metadata:
Training Memory (GB): 7.2
inference time (ms/im):
- value: 196.08
hardware: V100
backend: PyTorch
batch size: 1
mode: FP32
resolution: (800, 1333)
Epochs: 12
Results:
- Task: Object Detection
Dataset: COCO
Metrics:
box AP: 40.7
Weights: https://download.openmmlab.com/mmdetection/v2.0/ghm/retinanet_ghm_x101_32x4d_fpn_1x_coco/retinanet_ghm_x101_32x4d_fpn_1x_coco_20200131-e4333bd0.pth
- Name: retinanet_ghm_x101_64x4d_fpn_1x_coco
In Collection: GHM
Config: configs/ghm/retinanet_ghm_x101_64x4d_fpn_1x_coco.py
Metadata:
Training Memory (GB): 10.3
inference time (ms/im):
- value: 192.31
hardware: V100
backend: PyTorch
batch size: 1
mode: FP32
resolution: (800, 1333)
Epochs: 12
Results:
- Task: Object Detection
Dataset: COCO
Metrics:
box AP: 41.4
Weights: https://download.openmmlab.com/mmdetection/v2.0/ghm/retinanet_ghm_x101_64x4d_fpn_1x_coco/retinanet_ghm_x101_64x4d_fpn_1x_coco_20200131-dd381cef.pth