Skip to content

GitLab

Unverified Commit 055fd973 authored by Yezhen Cong's avatar Yezhen Cong Committed by GitHub
Browse files

[Enhance] Change Groupfree3D config (#855) 

* All mods

* PointSample

* PointSample
parent 90e63784
Show whitespace changes
Inline Side-by-side
Showing with 24 additions and 56 deletions
configs/groupfree3d/README.md
View file @ 055fd973
...@@ -21,10 +21,10 @@ We implement Group-Free-3D and provide the result and checkpoints on ScanNet dat ...@@ -21,10 +21,10 @@ We implement Group-Free-3D and provide the result and checkpoints on ScanNet dat
| Method | Backbone | Lr schd | Mem (GB) | Inf time (fps) | AP@0.25 |AP@0.5| Download | | Method | Backbone | Lr schd | Mem (GB) | Inf time (fps) | AP@0.25 |AP@0.5| Download |
| :------: | :---------: | :-----: | :------: | :------------: | :----: |:----: | :------: | | :------: | :---------: | :-----: | :------: | :------------: | :----: |:----: | :------: |
| [L6, O256](./groupfree3d_8x4_scannet-3d-18class-L6-O256.py ) | PointNet++ | 3x |6.7||65.59 (65.67*)|48.43 (47.74*)|[model](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-L6-O256/groupfree3d_8x4_scannet-3d-18class-L6-O256_20210702_145347-3499eb55.pth) | [log](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-L6-O256/groupfree3d_8x4_scannet-3d-18class-L6-O256_20210702_145347.log.json)| | [L6, O256](./groupfree3d_8x4_scannet-3d-18class-L6-O256.py ) | PointNet++ | 3x |6.7||66.32 (65.67*)|47.82 (47.74*)|[model](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-L6-O256/groupfree3d_8x4_scannet-3d-18class-L6-O256_20210702_145347-3499eb55.pth) | [log](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-L6-O256/groupfree3d_8x4_scannet-3d-18class-L6-O256_20210702_145347.log.json)|
| [L12, O256](./groupfree3d_8x4_scannet-3d-18class-L12-O256.py ) | PointNet++ | 3x |9.4||67.68 (66.22*)|49.30 (48.95*)|[model](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-L12-O256/groupfree3d_8x4_scannet-3d-18class-L12-O256_20210702_150907-1c5551ad.pth) | [log](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-L12-O256/groupfree3d_8x4_scannet-3d-18class-L12-O256_20210702_150907.log.json)| | [L12, O256](./groupfree3d_8x4_scannet-3d-18class-L12-O256.py ) | PointNet++ | 3x |9.4||66.57 (66.22*)|48.21 (48.95*)|[model](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-L12-O256/groupfree3d_8x4_scannet-3d-18class-L12-O256_20210702_150907-1c5551ad.pth) | [log](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-L12-O256/groupfree3d_8x4_scannet-3d-18class-L12-O256_20210702_150907.log.json)|
| [L12, O256](./groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256.py ) | PointNet++w2x | 3x |13.3||67.09 (67.30*)|50.76 (50.44*)|[model](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256_20210702_200301-944f0ac0.pth) | [log](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256_20210702_200301.log.json)| | [L12, O256](./groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256.py ) | PointNet++w2x | 3x |13.3||68.20 (67.30*)|51.02 (50.44*)|[model](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256_20210702_200301-944f0ac0.pth) | [log](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256_20210702_200301.log.json)|
| [L12, O512](./groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512.py ) | PointNet++w2x | 3x |18.8||68.31 (68.20*)|51.73 (51.31*)|[model](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512_20210702_220204-187b71c7.pth) | [log](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512_20210702_220204.log.json)| | [L12, O512](./groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512.py ) | PointNet++w2x | 3x |18.8||68.22 (68.20*)|52.61 (51.31*)|[model](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512_20210702_220204-187b71c7.pth) | [log](https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512_20210702_220204.log.json)|
**Notes:** **Notes:**
......
configs/groupfree3d/groupfree3d_8x4_scannet-3d-18class-L12-O256.py
View file @ 055fd973
...@@ -147,7 +147,7 @@ data = dict( ...@@ -147,7 +147,7 @@ data = dict(
workers_per_gpu=4, workers_per_gpu=4,
train=dict( train=dict(
type='RepeatDataset', type='RepeatDataset',
times=1, times=5,
dataset=dict( dataset=dict(
type=dataset_type, type=dataset_type,
data_root=data_root, data_root=data_root,
...@@ -192,16 +192,8 @@ optimizer = dict( ...@@ -192,16 +192,8 @@ optimizer = dict(
})) }))
optimizer_config = dict(grad_clip=dict(max_norm=0.1, norm_type=2)) optimizer_config = dict(grad_clip=dict(max_norm=0.1, norm_type=2))
lr_config = dict(policy='step', warmup=None, step=[280, 340]) lr_config = dict(policy='step', warmup=None, step=[56, 68])
# runtime settings # runtime settings
runner = dict(type='EpochBasedRunner', max_epochs=400) runner = dict(type='EpochBasedRunner', max_epochs=80)
checkpoint_config = dict(interval=1, max_keep_ckpts=10) checkpoint_config = dict(interval=1, max_keep_ckpts=10)
# yapf:disable
log_config = dict(
interval=30,
hooks=[
dict(type='TextLoggerHook'),
dict(type='TensorboardLoggerHook')
])
# yapf:enable
configs/groupfree3d/groupfree3d_8x4_scannet-3d-18class-L6-O256.py
View file @ 055fd973
...@@ -146,7 +146,7 @@ data = dict( ...@@ -146,7 +146,7 @@ data = dict(
workers_per_gpu=4, workers_per_gpu=4,
train=dict( train=dict(
type='RepeatDataset', type='RepeatDataset',
times=1, times=5,
dataset=dict( dataset=dict(
type=dataset_type, type=dataset_type,
data_root=data_root, data_root=data_root,
...@@ -191,16 +191,8 @@ optimizer = dict( ...@@ -191,16 +191,8 @@ optimizer = dict(
})) }))
optimizer_config = dict(grad_clip=dict(max_norm=0.1, norm_type=2)) optimizer_config = dict(grad_clip=dict(max_norm=0.1, norm_type=2))
lr_config = dict(policy='step', warmup=None, step=[280, 340]) lr_config = dict(policy='step', warmup=None, step=[56, 68])
# runtime settings # runtime settings
runner = dict(type='EpochBasedRunner', max_epochs=400) runner = dict(type='EpochBasedRunner', max_epochs=80)
checkpoint_config = dict(interval=1, max_keep_ckpts=10) checkpoint_config = dict(interval=1, max_keep_ckpts=10)
# yapf:disable
log_config = dict(
interval=30,
hooks=[
dict(type='TextLoggerHook'),
dict(type='TensorboardLoggerHook')
])
# yapf:enable
configs/groupfree3d/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256.py
View file @ 055fd973
...@@ -162,7 +162,7 @@ data = dict( ...@@ -162,7 +162,7 @@ data = dict(
workers_per_gpu=4, workers_per_gpu=4,
train=dict( train=dict(
type='RepeatDataset', type='RepeatDataset',
times=1, times=5,
dataset=dict( dataset=dict(
type=dataset_type, type=dataset_type,
data_root=data_root, data_root=data_root,
...@@ -207,16 +207,8 @@ optimizer = dict( ...@@ -207,16 +207,8 @@ optimizer = dict(
})) }))
optimizer_config = dict(grad_clip=dict(max_norm=0.1, norm_type=2)) optimizer_config = dict(grad_clip=dict(max_norm=0.1, norm_type=2))
lr_config = dict(policy='step', warmup=None, step=[280, 340]) lr_config = dict(policy='step', warmup=None, step=[56, 68])
# runtime settings # runtime settings
runner = dict(type='EpochBasedRunner', max_epochs=400) runner = dict(type='EpochBasedRunner', max_epochs=80)
checkpoint_config = dict(interval=1, max_keep_ckpts=10) checkpoint_config = dict(interval=1, max_keep_ckpts=10)
# yapf:disable
log_config = dict(
interval=30,
hooks=[
dict(type='TextLoggerHook'),
dict(type='TensorboardLoggerHook')
])
# yapf:enable
configs/groupfree3d/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512.py
View file @ 055fd973
...@@ -163,7 +163,7 @@ data = dict( ...@@ -163,7 +163,7 @@ data = dict(
workers_per_gpu=4, workers_per_gpu=4,
train=dict( train=dict(
type='RepeatDataset', type='RepeatDataset',
times=1, times=5,
dataset=dict( dataset=dict(
type=dataset_type, type=dataset_type,
data_root=data_root, data_root=data_root,
...@@ -208,16 +208,8 @@ optimizer = dict( ...@@ -208,16 +208,8 @@ optimizer = dict(
})) }))
optimizer_config = dict(grad_clip=dict(max_norm=0.1, norm_type=2)) optimizer_config = dict(grad_clip=dict(max_norm=0.1, norm_type=2))
lr_config = dict(policy='step', warmup=None, step=[280, 340]) lr_config = dict(policy='step', warmup=None, step=[56, 68])
# runtime settings # runtime settings
runner = dict(type='EpochBasedRunner', max_epochs=400) runner = dict(type='EpochBasedRunner', max_epochs=80)
checkpoint_config = dict(interval=1, max_keep_ckpts=10) checkpoint_config = dict(interval=1, max_keep_ckpts=10)
# yapf:disable
log_config = dict(
interval=30,
hooks=[
dict(type='TextLoggerHook'),
dict(type='TensorboardLoggerHook')
])
# yapf:enable
configs/groupfree3d/metafile.yml
View file @ 055fd973
...@@ -20,8 +20,8 @@ Models: ...@@ -20,8 +20,8 @@ Models:
- Task: 3D Object Detection - Task: 3D Object Detection
Dataset: ScanNet Dataset: ScanNet
Metrics: Metrics:
AP@0.25: 65.59 AP@0.25: 66.32
AP@0.5: 48.43 AP@0.5: 47.82
Weights: https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-L6-O256/groupfree3d_8x4_scannet-3d-18class-L6-O256_20210702_145347-3499eb55.pth Weights: https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-L6-O256/groupfree3d_8x4_scannet-3d-18class-L6-O256_20210702_145347-3499eb55.pth
- Name: groupfree3d_8x4_scannet-3d-18class-L12-O256.py - Name: groupfree3d_8x4_scannet-3d-18class-L12-O256.py
...@@ -34,8 +34,8 @@ Models: ...@@ -34,8 +34,8 @@ Models:
- Task: 3D Object Detection - Task: 3D Object Detection
Dataset: ScanNet Dataset: ScanNet
Metrics: Metrics:
AP@0.25: 67.68 AP@0.25: 66.57
AP@0.5: 49.30 AP@0.5: 48.21
Weights: https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-L12-O256/groupfree3d_8x4_scannet-3d-18class-L12-O256_20210702_150907-1c5551ad.pth Weights: https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-L12-O256/groupfree3d_8x4_scannet-3d-18class-L12-O256_20210702_150907-1c5551ad.pth
- Name: groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256.py - Name: groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256.py
...@@ -48,8 +48,8 @@ Models: ...@@ -48,8 +48,8 @@ Models:
- Task: 3D Object Detection - Task: 3D Object Detection
Dataset: ScanNet Dataset: ScanNet
Metrics: Metrics:
AP@0.25: 67.09 AP@0.25: 68.20
AP@0.5: 50.76 AP@0.5: 51.02
Weights: https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256_20210702_200301-944f0ac0.pth Weights: https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256_20210702_200301-944f0ac0.pth
- Name: groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512.py - Name: groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512.py
...@@ -62,6 +62,6 @@ Models: ...@@ -62,6 +62,6 @@ Models:
- Task: 3D Object Detection - Task: 3D Object Detection
Dataset: ScanNet Dataset: ScanNet
Metrics: Metrics:
AP@0.25: 68.31 AP@0.25: 68.22
AP@0.5: 51.73 AP@0.5: 52.61
Weights: https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512_20210702_220204-187b71c7.pth Weights: https://download.openmmlab.com/mmdetection3d/v0.1.0_models/groupfree3d/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512/groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512_20210702_220204-187b71c7.pth
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment