metafile.yml 1.28 KB
Newer Older
unknown's avatar
unknown committed
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
Collections:
  - Name: CenterNet
    Metadata:
      Training Data: COCO
      Training Techniques:
        - SGD with Momentum
        - Weight Decay
      Training Resources: 8x NVIDIA V100 GPUs
      Architecture:
        - ResNet
    Paper: https://arxiv.org/abs/1904.07850
    README: configs/centernet/README.md

Models:
  - Name: centernet_resnet18_dcnv2_140e_coco
    In Collection: CenterNet
    Config: configs/centernet/centernet_resnet18_dcnv2_140e_coco.py
    Metadata:
      Batch Size: 128
      Training Memory (GB): 3.47
      Epochs: 140
    Results:
      - Task: Object Detection
        Dataset: COCO
        Metrics:
          box AP: 29.5
    Weights: https://download.openmmlab.com/mmdetection/v2.0/centernet/centernet_resnet18_dcnv2_140e_coco/centernet_resnet18_dcnv2_140e_coco_20210520_101209-da388ba2.pth

  - Name: centernet_resnet18_140e_coco
    In Collection: CenterNet
    Config: configs/centernet/centernet_resnet18_140e_coco.py
    Metadata:
      Batch Size: 128
      Training Memory (GB): 3.45
      Epochs: 140
    Results:
      - Task: Object Detection
        Dataset: COCO
        Metrics:
          box AP: 26.0
    Weights: https://download.openmmlab.com/mmdetection/v2.0/centernet/centernet_resnet18_140e_coco/centernet_resnet18_140e_coco_20210519_092334-eafe8ccd.pth