forked from open-mmlab/mmdetection
-
Notifications
You must be signed in to change notification settings - Fork 0
/
metafile.yml
78 lines (75 loc) · 2.24 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
Collections:
- Name: SSD
Metadata:
Training Data: COCO
Training Techniques:
- SGD with Momentum
- Weight Decay
Training Resources: 8x V100 GPUs
Architecture:
- VGG
Paper:
URL: https://arxiv.org/abs/1512.02325
Title: 'SSD: Single Shot MultiBox Detector'
README: configs/ssd/README.md
Code:
URL: https://github.com/open-mmlab/mmdetection/blob/v2.14.0/mmdet/models/dense_heads/ssd_head.py#L16
Version: v2.14.0
Models:
- Name: ssd300_coco
In Collection: SSD
Config: configs/ssd/ssd300_coco.py
Metadata:
Training Memory (GB): 9.9
inference time (ms/im):
- value: 22.88
hardware: V100
backend: PyTorch
batch size: 1
mode: FP32
resolution: (300, 300)
Epochs: 120
Results:
- Task: Object Detection
Dataset: COCO
Metrics:
box AP: 25.5
Weights: https://download.openmmlab.com/mmdetection/v2.0/ssd/ssd300_coco/ssd300_coco_20210803_015428-d231a06e.pth
- Name: ssd512_coco
In Collection: SSD
Config: configs/ssd/ssd512_coco.py
Metadata:
Training Memory (GB): 19.4
inference time (ms/im):
- value: 32.57
hardware: V100
backend: PyTorch
batch size: 1
mode: FP32
resolution: (512, 512)
Epochs: 120
Results:
- Task: Object Detection
Dataset: COCO
Metrics:
box AP: 29.5
Weights: https://download.openmmlab.com/mmdetection/v2.0/ssd/ssd512_coco/ssd512_coco_20210803_022849-0a47a1ca.pth
- Name: ssdlite_mobilenetv2-scratch_8xb24-600e_coco
In Collection: SSD
Config: configs/ssd/ssdlite_mobilenetv2-scratch_8xb24-600e_coco.py
Metadata:
Training Memory (GB): 4.0
inference time (ms/im):
- value: 14.3
hardware: V100
backend: PyTorch
batch size: 1
mode: FP32
resolution: (320, 320)
Epochs: 600
Results:
- Task: Object Detection
Dataset: COCO
Metrics:
box AP: 21.3
Weights: https://download.openmmlab.com/mmdetection/v2.0/ssd/ssdlite_mobilenetv2_scratch_600e_coco/ssdlite_mobilenetv2_scratch_600e_coco_20210629_110627-974d9307.pth