search

TRAILab / CaDDN

Categorical Depth Distribution Network for Monocular 3D Object Detection (CVPR 2021 Oral)
Apache License 2.0
359 stars 62 forks source link

ImportError: cannot import name 'iou3d_nms_cuda' from 'pcdet.ops.iou3d_nms' (unknown location) #52

Closed zjyalice closed 3 years ago

zjyalice commented 3 years ago

Hi~

Thank you very much for sharing your great work with us.

When I generate data info in kitti format, the following error occurred: ImportError: cannot import name 'iou3d_nms_cuda' from 'pcdet.ops.iou3d_nms' (unknown location)

Then I search for 'iou3d_nms_cuda' file in 'pcdet.ops.iou3d_nms' , but it is not found. Is this file missing?How should I solve this problem?

My dataset is in KITTI format, and I execute the following code: python -m pcdet.datasets.kitti.kitti_dataset create_kitti_infos tools/cfgs/dataset_configs/kitti_dataset.yaml

I would really appreciate if you could help.

Best.

zjyalice commented 3 years ago

Sorry, there's something wrong with my conda environment. Just reconfigure the environment.

shiqi1994 commented 2 years ago

That should be a problem on spconv compiling. I have this problem too.

Rinatum commented 2 years ago

@shiqi1994 Did you solve this problem? I use docker image and have same error for from pcdet.datasets import build_dataloader