open-mmlab / OpenPCDet

OpenPCDet Toolbox for LiDAR-based 3D Object Detection.
Apache License 2.0
4.61k stars 1.29k forks source link

CADDN demo run problem #819

Closed ghost closed 2 years ago

ghost commented 2 years ago

I am getting this error File "/media/safa/nvme/pycharmProjects/OpenPCDet/pcdet/datasets/processor/data_processor.py", line 195, in downsample_depth_map image=data_dict['depth_maps'], KeyError: 'depth_maps'

When I run

python demo.py --cfg_file cfgs/kitti_models/CaDDN.yaml --ckpt caddn_pcdet.pth --data_path ../data/kitti/training/velodyne/000008.bin

I have downloaded depth_2 folder near calib and image_2 folder in train dataset.

What am I doing wrong?

Regards

Safa

MartinHahner commented 2 years ago

Have you generated the .pkl files as described here before after you downloaded the depth maps?

Maybe you need to re-generate them if you downloaded the depth maps afterwards. Otherwise, perhaps @codyreading can help.

ghost commented 2 years ago

Thank you @MartinHahner for the reply. Stating pkl file generation, I guess you mean running this command below:

python -m pcdet.datasets.kitti.kitti_dataset create_kitti_infos tools/cfgs/dataset_configs/kitti_dataset.yaml I had done before I downloaded. I did after I downloaded, per your advice. Still I am getting the same error.

Traceback (most recent call last):
  File "demo.py", line 113, in <module>
    main()
  File "demo.py", line 95, in main
    for idx, data_dict in enumerate(demo_dataset):
  File "demo.py", line 60, in __getitem__
    data_dict = self.prepare_data(data_dict=input_dict)
  File "/media/safa/nvme/pycharmProjects/OpenPCDet/pcdet/datasets/dataset.py", line 148, in prepare_data
    data_dict = self.data_processor.forward(
  File "/media/safa/nvme/pycharmProjects/OpenPCDet/pcdet/datasets/processor/data_processor.py", line 213, in forward
    data_dict = cur_processor(data_dict=data_dict)
  File "/media/safa/nvme/pycharmProjects/OpenPCDet/pcdet/datasets/processor/data_processor.py", line 195, in downsample_depth_map
    image=data_dict['depth_maps'],
KeyError: 'depth_maps'

I am open to other suggestions as well.

Regards

creading7 commented 2 years ago

Hi @safaATcurrus,

I never updated the demo script to work for CaDDN. The issue is that the DemoDataset object doesn't load the depth maps and images, so you will need to manually add those similar to the kitti_dataset.py. Checkout here and here for how to load the images and depth maps respectively.

github-actions[bot] commented 2 years ago

This issue is stale because it has been open for 30 days with no activity.

github-actions[bot] commented 2 years ago

This issue was closed because it has been inactive for 14 days since being marked as stale.

Mi97ch commented 1 year ago

Hi, do someone have the updated Demo code to solve this issue?