Skip to content

Commit

Permalink
[Fix] Refine metafiles (#690)
Browse files Browse the repository at this point in the history
* [Fix] Refine metafiles

* fix MANIFEST.in
  • Loading branch information
wHao-Wu authored Jul 2, 2021
1 parent 653aa75 commit f14b2fe
Show file tree
Hide file tree
Showing 4 changed files with 66 additions and 1 deletion.
2 changes: 1 addition & 1 deletion MANIFEST.in
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
include mmdet3d/model_zoo.yml
include mmdet3d/model-index.yml
include requirements/*.txt
recursive-include mmdet3d/configs *.py *.yml
recursive-include mmdet3d/tools *.sh *.py
Expand Down
38 changes: 38 additions & 0 deletions configs/fcos3d/metafile.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
Collections:
- Name: FCOS3D
Metadata:
Training Data: NuScenes
Training Techniques:
- SGD
Training Resources: 8x GeForce RTX 2080 Ti
Architecture:
- FCOSMono3DHead
Paper: https://arxiv.org/abs/2104.10956
README: configs/fcos3d/README.md

Models:
- Name: fcos3d_r101_caffe_fpn_gn-head_dcn_2x8_1x_nus-mono3d
In Collection: FCOS3D
Config: configs/fcos3d/fcos3d_r101_caffe_fpn_gn-head_dcn_2x8_1x_nus-mono3d.py
Metadata:
Training Memory (GB): 8.7
Results:
- Task: 3D Object Detection
Dataset: NuScenes
Metrics:
mAP: 29.9
NDS: 37.3
Weights: https://download.openmmlab.com/mmdetection3d/v0.1.0_models/fcos3d/fcos3d_r101_caffe_fpn_gn-head_dcn_2x8_1x_nus-mono3d/fcos3d_r101_caffe_fpn_gn-head_dcn_2x8_1x_nus-mono3d_20210425_181341-8d5a21fe.pth

- Name: fcos3d_r101_caffe_fpn_gn-head_dcn_2x8_1x_nus-mono3d_finetune
In Collection: FCOS3D
Config: configs/fcos3d/fcos3d_r101_caffe_fpn_gn-head_dcn_2x8_1x_nus-mono3d_finetune.py
Metadata:
Training Memory (GB): 8.7
Results:
- Task: 3D Object Detection
Dataset: NuScenes
Metrics:
mAP: 32.1
NDS: 39.3
Weights: https://download.openmmlab.com/mmdetection3d/v0.1.0_models/fcos3d/fcos3d_r101_caffe_fpn_gn-head_dcn_2x8_1x_nus-mono3d_finetune/fcos3d_r101_caffe_fpn_gn-head_dcn_2x8_1x_nus-mono3d_finetune_20210427_091419-35aaaad0.pth
24 changes: 24 additions & 0 deletions configs/imvoxelnet/metafile.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
Collections:
- Name: ImVoxelNet
Metadata:
Training Data: KITTI
Training Techniques:
- AdamW
Training Resources: 8x Tesla P40
Architecture:
- Anchor3DHead
Paper: https://arxiv.org/abs/2106.01178
README: configs/imvoxelnet/README.md

Models:
- Name: imvoxelnet_kitti-3d-car
In Collection: ImVoxelNet
Config: configs/imvoxelnet/imvoxelnet_kitti-3d-car.py
Metadata:
Training Memory (GB): 15.0
Results:
- Task: 3D Object Detection
Dataset: KITTI
Metrics:
mAP: 17.4
Weights: https://download.openmmlab.com/mmdetection3d/v0.1.0_models/imvoxelnet/imvoxelnet_kitti-3d-car_20210610_152323-b9abba85.pth
3 changes: 3 additions & 0 deletions model_zoo.yml → model-index.yml
Original file line number Diff line number Diff line change
Expand Up @@ -9,8 +9,11 @@ Import:
- configs/mvxnet/metafile.yml
- configs/nuimages/metafile.yml
- configs/parta2/metafile.yml
- configs/pointnet2/metafile.yml
- configs/pointpillars/metafile.yml
- configs/regnet/metafile.yml
- configs/second/metafile.yml
- configs/ssn/metafile.yml
- configs/votenet/metafile.yml
- configs/fcos3d/metafile.yml
- configs/imvoxelnet/metafile.yml

0 comments on commit f14b2fe

Please sign in to comment.