question about preprocess nuscenes datasets

[jiumozhi123]

Hello, I preprocess nuscenes datasets by "python tools/create_data.py nuscenes --root-path ./data/nuscenes --out-dir ./data/nuscenes --extra-tag nuscenes --version v1.0".The following error occurred:

My python environment is: cuda 11.1, python 3.7.15, pytorch1.7.1, mmcv-full1.4.0, mmdet2.14.0, mmdet3d0.16.0(in the current project)
Related code is:
`def nuscenes_data_prep(root_path,
info_prefix,
version,
dataset_name,
out_dir,
max_sweeps=10):
nuscenes_converter.create_nuscenes_infos(
root_path, info_prefix, version=version, max_sweeps=max_sweeps)

if version == 'v1.0-test':
    info_test_path = osp.join(root_path, f'{info_prefix}_infos_test.pkl')
    nuscenes_converter.export_2d_annotation(
        root_path, info_test_path, version=version)
    return

info_train_path = osp.join(root_path, f'{info_prefix}_infos_train.pkl')
info_val_path = osp.join(root_path, f'{info_prefix}_infos_val.pkl')
# nuscenes_converter.export_2d_annotation(
#     root_path, info_train_path, version=version)
# nuscenes_converter.export_2d_annotation(
#     root_path, info_val_path, version=version)
create_groundtruth_database(dataset_name, root_path, info_prefix,
    f'{out_dir}/{info_prefix}_infos_train.pkl',
    with_bbox=True)`

Could you help to analyze the cause of the problem?
Thanks

[chaoyiwang09]

I also come up with this problem.

[ZecCheng]

Hello, I preprocess nuscenes datasets by "python tools/create_data.py nuscenes --root-path ./data/nuscenes --out-dir ./data/nuscenes --extra-tag nuscenes --version v1.0".The following error occurred:

My python environment is: cuda 11.1, python 3.7.15, pytorch1.7.1, mmcv-full1.4.0, mmdet2.14.0, mmdet3d0.16.0(in the current project) Related code is: `def nuscenes_data_prep(root_path, info_prefix, version, dataset_name, out_dir, max_sweeps=10): nuscenes_converter.create_nuscenes_infos( root_path, info_prefix, version=version, max_sweeps=max_sweeps)

if version == 'v1.0-test':
    info_test_path = osp.join(root_path, f'{info_prefix}_infos_test.pkl')
    nuscenes_converter.export_2d_annotation(
        root_path, info_test_path, version=version)
    return

info_train_path = osp.join(root_path, f'{info_prefix}_infos_train.pkl')
info_val_path = osp.join(root_path, f'{info_prefix}_infos_val.pkl')
# nuscenes_converter.export_2d_annotation(
#     root_path, info_train_path, version=version)
# nuscenes_converter.export_2d_annotation(
#     root_path, info_val_path, version=version)
create_groundtruth_database(dataset_name, root_path, info_prefix,
    f'{out_dir}/{info_prefix}_infos_train.pkl',
    with_bbox=True)`

Could you help to analyze the cause of the problem? Thanks

Hey, did you solve the problem? Is is the mmdet3d version problem or others?

[zehuichen123]

You may put a pdb and print the dict name in the example['img_info'], and check the dict value is filename or file_name or other values. Simply changing this could be ok.

[ZecCheng]

"My python environment is: cuda 11.1, python 3.7.15, pytorch1.7.1, mmcv-full1.4.0, mmdet2.14.0, mmdet3d0.16.0(in the current project)"
Hey,zehui, is this environment works well for your project? Actually, I wonder which mmdet3d should I install because there are updates between different versions. 😄

[zehuichen123]

LGTM.