Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Fix] Refine metafiles #690

Merged
merged 2 commits into from
Jul 2, 2021
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
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