metafile.yml 2.69 KB
Newer Older
1
2
3
4
5
6
7
8
Collections:
  - Name: SSN
    Metadata:
      Training Techniques:
        - AdamW
      Training Resources: 8x GeForce GTX 1080 Ti
      Architecture:
        - Hard Voxelization
dingchang's avatar
dingchang committed
9
10
11
    Paper:
      URL: https://arxiv.org/abs/2004.02774
      Title: 'SSN: Shape Signature Networks for Multi-class Object Detection from Point Clouds'
12
    README: configs/ssn/README.md
dingchang's avatar
dingchang committed
13
14
15
    Code:
      URL: https://github.com/open-mmlab/mmdetection3d/blob/master/mmdet3d/models/dense_heads/shape_aware_head.py#L166
      Version: v0.7.0
16
17

Models:
18
  - Name: hv_ssn_secfpn_sbn-all_16xb2-2x_nus-3d
19
    In Collection: SSN
20
    Config: configs/ssn/ssn_hv_secfpn_sbn-all_16xb2-2x_nus-3d.py
21
22
    Metadata:
      Training Data: nuScenes
23
      Training Memory (GB): 3.6
24
25
26
27
    Results:
      - Task: 3D Object Detection
        Dataset: nuScenes
        Metrics:
28
29
30
          mAP: 40.91
          NDS: 54.44
    Weights: https://download.openmmlab.com/mmdetection3d/v1.0.0_models/ssn/hv_ssn_secfpn_sbn-all_2x16_2x_nus-3d/hv_ssn_secfpn_sbn-all_2x16_2x_nus-3d_20210830_101351-51915986.pth
31

32
  - Name: hv_ssn_regnet-400mf_secfpn_sbn-all_16xb2-2x_nus-3d
33
    In Collection: SSN
34
    Config: configs/ssn/ssn_hv_regnet-400mf_secfpn_sbn-all_16xb2-2x_nus-3d.py
35
36
    Metadata:
      Training Data: nuScenes
37
      Training Memory (GB): 5.1
38
39
40
41
    Results:
      - Task: 3D Object Detection
        Dataset: nuScenes
        Metrics:
42
          mAP: 46.65
43
          NDS: 58.24
44
    Weights: https://download.openmmlab.com/mmdetection3d/v1.0.0_models/ssn/hv_ssn_regnet-400mf_secfpn_sbn-all_2x16_2x_nus-3d/hv_ssn_regnet-400mf_secfpn_sbn-all_2x16_2x_nus-3d_20210829_210615-361e5e04.pth
45

46
  - Name: hv_ssn_secfpn_sbn-all_16xb2-2x_lyft-3d
47
    In Collection: SSN
48
    Config: configs/ssn/ssn_hv_secfpn_sbn-all_16xb2-2x_lyft-3d.py
49
50
    Metadata:
      Training Data: Lyft
51
      Training Memory (GB): 8.5
52
53
54
55
    Results:
      - Task: 3D Object Detection
        Dataset: Lyft
        Metrics:
56
          Private Score: 17.5
57
          Public Score: 17.5
58
    Weights: https://download.openmmlab.com/mmdetection3d/v1.0.0_models/ssn/hv_ssn_secfpn_sbn-all_2x16_2x_lyft-3d/hv_ssn_secfpn_sbn-all_2x16_2x_lyft-3d_20210822_134731-46841b41.pth
59

60
  - Name: hv_ssn_regnet-400mf_secfpn_sbn-all_16xb1-2x_lyft-3d
61
    In Collection: SSN
62
    Config: configs/ssn/hv_ssn_regnet-400mf_secfpn_sbn-all_16xb1-2x_lyft-3d.py
63
64
    Metadata:
      Training Data: Lyft
65
      Training Memory (GB): 7.4
66
67
68
69
    Results:
      - Task: 3D Object Detection
        Dataset: Lyft
        Metrics:
70
71
72
          Private Score: 17.9
          Public Score: 18.0
    Weights: https://download.openmmlab.com/mmdetection3d/v1.0.0_models/ssn/hv_ssn_regnet-400mf_secfpn_sbn-all_1x16_2x_lyft-3d/hv_ssn_regnet-400mf_secfpn_sbn-all_1x16_2x_lyft-3d_20210829_122825-d93475a1.pth