Unverified Commit aecc7a17 authored by ZLTJohn's avatar ZLTJohn Committed by GitHub
Browse files

[Fix] fix detr3d 'custom_import' error (#2260)

* fix 'custom_imports' in config files

* fix lint
parent 377597b6
...@@ -3,7 +3,7 @@ _base_ = [ ...@@ -3,7 +3,7 @@ _base_ = [
'mmdet3d::_base_/default_runtime.py' 'mmdet3d::_base_/default_runtime.py'
] ]
custom_imports = dict(imports=['projects.detr3d.detr3d']) custom_imports = dict(imports=['projects.DETR3D.detr3d'])
# If point cloud range is changed, the models should also change their point # If point cloud range is changed, the models should also change their point
# cloud range accordingly # cloud range accordingly
point_cloud_range = [-51.2, -51.2, -5.0, 51.2, 51.2, 3.0] point_cloud_range = [-51.2, -51.2, -5.0, 51.2, 51.2, 3.0]
......
_base_ = ['./detr3d_r101_gridmask.py'] _base_ = ['./detr3d_r101_gridmask.py']
custom_imports = dict(imports=['projects.detr3d.detr3d']) custom_imports = dict(imports=['projects.DETR3D.detr3d'])
# If point cloud range is changed, the models should also change their point # If point cloud range is changed, the models should also change their point
# cloud range accordingly # cloud range accordingly
point_cloud_range = [-51.2, -51.2, -5.0, 51.2, 51.2, 3.0] point_cloud_range = [-51.2, -51.2, -5.0, 51.2, 51.2, 3.0]
......
_base_ = ['./detr3d_r101_gridmask_cbgs.py'] _base_ = ['./detr3d_r101_gridmask_cbgs.py']
custom_imports = dict(imports=['projects.detr3d.detr3d']) custom_imports = dict(imports=['projects.DETR3D.detr3d'])
img_norm_cfg = dict( img_norm_cfg = dict(
mean=[103.530, 116.280, 123.675], mean=[103.530, 116.280, 123.675],
...@@ -35,5 +35,17 @@ train_dataloader = dict( ...@@ -35,5 +35,17 @@ train_dataloader = dict(
dataset=dict( dataset=dict(
type='CBGSDataset', type='CBGSDataset',
dataset=dict(ann_file='nuscenes_infos_trainval.pkl'))) dataset=dict(ann_file='nuscenes_infos_trainval.pkl')))
test_dataloader = dict(
dataset=dict(
data_root='data/nuscenes-test', ann_file='nuscenes_infos_test.pkl'))
test_evaluator = dict(
type='NuScenesMetric',
data_root='data/nuscenes-test',
jsonfile_prefix='work_dirs/detr3d_vovnet_results_test',
format_only=True,
metric=[])
load_from = 'ckpts/dd3d_det_final.pth' load_from = 'ckpts/dd3d_det_final.pth'
find_unused_parameters = True find_unused_parameters = True
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