data preparation error #5
Description
First of all, thank you for sharing your wonderful work!
After completing the environment configuration, I executed the command following the description in the readme:
python tools/create_data.py nuscenes --root-path ./data/nuscenes --out-dir ./data/nuscenes --extra-tag nuscenes
Then, the following error is reported：
(alocc) Server:~/workspace/ALOcc$ python tools/create_data.py nuscenes --version v1.0-mini --root-path ./data/nuscenes --out-dir ./data/nuscenes --extra-tag nuscenes
/home/pc/workspace/ALOcc/mmdet3d/core/evaluation/kitti_utils/eval.py:10: NumbaDeprecationWarning: The 'nopython' keyword argument was not supplied to the 'numba.jit' decorator. The implicit default value for this argument is currently False, but it will be changed to True in Numba 0.59.0. See https://numba.readthedocs.io/en/stable/reference/deprecation.html#deprecation-of-object-mode-fall-back-behaviour-when-using-jit for details.
def get_thresholds(scores: np.ndarray, num_gt, num_sample_pts=41):
Using /home/pc/.cache/torch_extensions/py38_cu118 as PyTorch extensions root...
Detected CUDA files, patching ldflags
Emitting ninja build file /home/pc/.cache/torch_extensions/py38_cu118/dvr/build.ninja...
Building extension module dvr...
Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N)
ninja: no work to do.
Loading extension module dvr...
Loading NuScenes tables for version v1.0-mini...
Loading nuScenes-lidarseg...
32 category,
8 attribute,
4 visibility,
911 instance,
12 sensor,
120 calibrated_sensor,
31206 ego_pose,
8 log,
10 scene,
404 sample,
31206 sample_data,
18538 sample_annotation,
4 map,
404 lidarseg,
Done loading in 0.420 seconds.
Reverse indexing ...
Done reverse indexing in 0.1 seconds.
total scene num: 10
exist scene num: 10
train scene: 8, val scene: 2
[>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>] 404/404, 17.4 task/s, elapsed: 23s, ETA: 0s
train sample: 323, val sample: 81
Loading NuScenes tables for version v1.0-mini...
Loading nuScenes-lidarseg...
32 category,
8 attribute,
4 visibility,
911 instance,
12 sensor,
120 calibrated_sensor,
31206 ego_pose,
8 log,
10 scene,
404 sample,
31206 sample_data,
18538 sample_annotation,
4 map,
404 lidarseg,
Done loading in 0.349 seconds.
Reverse indexing ...
Done reverse indexing in 0.1 seconds.
[>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>] 323/323, 9.0 task/s, elapsed: 36s, ETA: 0s
Loading NuScenes tables for version v1.0-mini...
Loading nuScenes-lidarseg...
32 category,
8 attribute,
4 visibility,
911 instance,
12 sensor,
120 calibrated_sensor,
31206 ego_pose,
8 log,
10 scene,
404 sample,
31206 sample_data,
18538 sample_annotation,
4 map,
404 lidarseg,
Done loading in 0.347 seconds.
Reverse indexing ...
Done reverse indexing in 0.1 seconds.
[>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>] 81/81, 7.4 task/s, elapsed: 11s, ETA: 0s
Create GT Database of NuScenesDataset
[ ] 0/41, elapsed: 0s, ETA:Traceback (most recent call last):
File "tools/create_data.py", line 267, in
nuscenes_data_prep(
File "tools/create_data.py", line 89, in nuscenes_data_prep
create_groundtruth_database(dataset_name, root_path, info_prefix,
File "/home/pc/workspace/ALOcc/tools/data_converter/create_gt_database.py", line 287, in create_groundtruth_database
input_dict = dataset.get_data_info(j)
File "/home/pc/workspace/ALOcc/mmdet3d/datasets/nuscenes_dataset.py", line 496, in get_data_info
scene_name=info['scene_name'],
KeyError: 'scene_name'
Based on the code analysis, it seems that there are incompleteness in the open-source code. Is that correct? Looking forward to your reply。 Thank you so much!