search

LittlePey / SFD

Sparse Fuse Dense: Towards High Quality 3D Detection with Depth Completion (CVPR 2022, Oral)
Apache License 2.0
263 stars 35 forks source link

kitti info pickle file generation #18

Open zijianh4 opened 2 years ago

zijianh4 commented 2 years ago

Hi @LittlePey , Thanks for your great work. Just wonder if there is any script generating the info pickle files? Thanks! :)

LittlePey commented 2 years ago

Hi, you can run this script to generate info pickle files.

zijianh4 commented 2 years ago

Hi, you can run this script to generate info pickle files.

Hi, could you please provide the details of running the script, maybe the command? It seems that running it under the pcdet folder directly trigger some errors, like this:

image

Thanks!

zijianh4 commented 2 years ago

Hi, you can run this script to generate info pickle files.

Hi, I have resolved the problem above, but another error message comes up, which seems related to the generated pickle files. image

Can I just use USE_ROAD_PLANE: False in /SFD/tools/cfgs/kitti_models/sfd.yaml, or is there something I missed? Thanks!

LittlePey commented 2 years ago

Hi, you can use the following command:

python -m pcdet.datasets.kitti.kitti_dataset_sfd create_kitti_infos tools/cfgs/dataset_configs/kitti_dataset_sfd.yaml

And it seems that you did not set 'training=False' when initializing KittiDatasetSFD as us. When training=False, you will not run into this line: image

zijianh4 commented 2 years ago

Hi, you can use the following command:

python -m pcdet.datasets.kitti.kitti_dataset_sfd create_kitti_infos tools/cfgs/dataset_configs/kitti_dataset_sfd.yaml

And it seems that you did not set 'training=False' when initializing KittiDatasetSFD as us. When training=False, you will not run into this line: image

Hi, thanks! Does it mean that I need to change this line https://github.com/LittlePey/SFD/blob/03f83de462033e4f840f38f0cead44c506b3f1c5/tools/train.py#L112 to False? Thanks!

Dowe-dong commented 2 years ago

you can download the planes file from google. and this bug will disapper.