Skip to content

Commit

Permalink
Update the info path for the CVPR 2024 challenge
Browse files Browse the repository at this point in the history
  • Loading branch information
@Tai-Wang
Tai-Wang committed Mar 5, 2024
1 parent d51c772 commit b579296
Show file tree
Hide file tree
Showing 5 changed files with 72 additions and 80 deletions.
4 changes: 2 additions & 2 deletions configs/detection/cont-det3d_8xb1_embodiedscan-3d-284class-9dof.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -190,7 +190,7 @@
times=8,
dataset=dict(type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_train_split.pkl',
ann_file='embodiedscan_infos_train.pkl',
pipeline=train_pipeline,
test_mode=False,
filter_empty_gt=True,
Expand All @@ -205,7 +205,7 @@
sampler=dict(type='DefaultSampler', shuffle=False),
dataset=dict(type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_val_split.pkl',
ann_file='embodiedscan_infos_val.pkl',
pipeline=test_pipeline,
test_mode=True,
filter_empty_gt=True,
Expand Down
34 changes: 16 additions & 18 deletions configs/detection/mv-det3d_8xb4_embodiedscan-3d-284class-9dof.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -184,30 +184,28 @@
sampler=dict(type='DefaultSampler', shuffle=True),
dataset=dict(type='RepeatDataset',
times=10,
dataset=dict(
type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_train_split_filtered.pkl',
pipeline=train_pipeline,
test_mode=False,
filter_empty_gt=True,
box_type_3d='Euler-Depth',
metainfo=metainfo)))
dataset=dict(type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_train.pkl',
pipeline=train_pipeline,
test_mode=False,
filter_empty_gt=True,
box_type_3d='Euler-Depth',
metainfo=metainfo)))

val_dataloader = dict(batch_size=1,
num_workers=1,
persistent_workers=True,
drop_last=False,
sampler=dict(type='DefaultSampler', shuffle=False),
dataset=dict(
type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_val_split_filtered.pkl',
pipeline=test_pipeline,
test_mode=True,
filter_empty_gt=True,
box_type_3d='Euler-Depth',
metainfo=metainfo))
dataset=dict(type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_val.pkl',
pipeline=test_pipeline,
test_mode=True,
filter_empty_gt=True,
box_type_3d='Euler-Depth',
metainfo=metainfo))
test_dataloader = val_dataloader

val_evaluator = dict(type='IndoorDetMetric')
Expand Down
38 changes: 18 additions & 20 deletions configs/grounding/mv-grounding_8xb12_embodiedscan-vg-9dof-full.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -139,32 +139,30 @@
sampler=dict(type='DefaultSampler', shuffle=True),
dataset=dict(type='RepeatDataset',
times=1,
dataset=dict(
type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_train_split_filtered.pkl',
vg_file='embodiedscan_train_full_vg.json',
metainfo=metainfo,
pipeline=train_pipeline,
test_mode=False,
filter_empty_gt=True,
box_type_3d='Euler-Depth')))
dataset=dict(type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_train.pkl',
vg_file='embodiedscan_train_full_vg.json',
metainfo=metainfo,
pipeline=train_pipeline,
test_mode=False,
filter_empty_gt=True,
box_type_3d='Euler-Depth')))

val_dataloader = dict(batch_size=12,
num_workers=12,
persistent_workers=True,
drop_last=False,
sampler=dict(type='DefaultSampler', shuffle=False),
dataset=dict(
type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_val_split_filtered.pkl',
vg_file='embodiedscan_val_full_vg.json',
metainfo=metainfo,
pipeline=test_pipeline,
test_mode=True,
filter_empty_gt=True,
box_type_3d='Euler-Depth'))
dataset=dict(type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_val.pkl',
vg_file='embodiedscan_val_full_vg.json',
metainfo=metainfo,
pipeline=test_pipeline,
test_mode=True,
filter_empty_gt=True,
box_type_3d='Euler-Depth'))
test_dataloader = val_dataloader

val_evaluator = dict(type='GroundingMetric')
Expand Down
38 changes: 18 additions & 20 deletions configs/grounding/mv-grounding_8xb12_embodiedscan-vg-9dof.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -139,32 +139,30 @@
sampler=dict(type='DefaultSampler', shuffle=True),
dataset=dict(type='RepeatDataset',
times=1,
dataset=dict(
type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_train_split_filtered.pkl',
vg_file='embodiedscan_train_vg.json',
metainfo=metainfo,
pipeline=train_pipeline,
test_mode=False,
filter_empty_gt=True,
box_type_3d='Euler-Depth')))
dataset=dict(type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_train.pkl',
vg_file='embodiedscan_train_mini_vg.json',
metainfo=metainfo,
pipeline=train_pipeline,
test_mode=False,
filter_empty_gt=True,
box_type_3d='Euler-Depth')))

val_dataloader = dict(batch_size=12,
num_workers=12,
persistent_workers=True,
drop_last=False,
sampler=dict(type='DefaultSampler', shuffle=False),
dataset=dict(
type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_val_split_filtered.pkl',
vg_file='embodiedscan_val_vg.json',
metainfo=metainfo,
pipeline=test_pipeline,
test_mode=True,
filter_empty_gt=True,
box_type_3d='Euler-Depth'))
dataset=dict(type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_val.pkl',
vg_file='embodiedscan_val_mini_vg.json',
metainfo=metainfo,
pipeline=test_pipeline,
test_mode=True,
filter_empty_gt=True,
box_type_3d='Euler-Depth'))
test_dataloader = val_dataloader

val_evaluator = dict(type='GroundingMetric')
Expand Down
38 changes: 18 additions & 20 deletions configs/grounding/mv-grounding_8xb12_embodiedscan-vg-9dof_fcaf-coder.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -140,32 +140,30 @@
sampler=dict(type='DefaultSampler', shuffle=True),
dataset=dict(type='RepeatDataset',
times=1,
dataset=dict(
type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_train_split_filtered.pkl',
vg_file='embodiedscan_train_vg.json',
metainfo=metainfo,
pipeline=train_pipeline,
test_mode=False,
filter_empty_gt=True,
box_type_3d='Euler-Depth')))
dataset=dict(type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_train.pkl',
vg_file='embodiedscan_train_mini_vg.json',
metainfo=metainfo,
pipeline=train_pipeline,
test_mode=False,
filter_empty_gt=True,
box_type_3d='Euler-Depth')))

val_dataloader = dict(batch_size=12,
num_workers=12,
persistent_workers=True,
drop_last=False,
sampler=dict(type='DefaultSampler', shuffle=False),
dataset=dict(
type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_val_split_filtered.pkl',
vg_file='embodiedscan_val_vg.json',
metainfo=metainfo,
pipeline=test_pipeline,
test_mode=True,
filter_empty_gt=True,
box_type_3d='Euler-Depth'))
dataset=dict(type=dataset_type,
data_root=data_root,
ann_file='embodiedscan_infos_val.pkl',
vg_file='embodiedscan_val_mini_vg.json',
metainfo=metainfo,
pipeline=test_pipeline,
test_mode=True,
filter_empty_gt=True,
box_type_3d='Euler-Depth'))
test_dataloader = val_dataloader

val_evaluator = dict(type='GroundingMetric')
Expand Down

0 comments on commit b579296

Please sign in to comment.