Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ModuleNotFoundError: No module named 'iou3d_nms_cuda' #9

Open
jackyyang-cv opened this issue Aug 12, 2021 · 2 comments
Open

ModuleNotFoundError: No module named 'iou3d_nms_cuda' #9

jackyyang-cv opened this issue Aug 12, 2021 · 2 comments

Comments

@jackyyang-cv
Copy link

Traceback (most recent call last):
File "train.py", line 19, in
from second.builder import target_assigner_builder, voxel_builder
File "/home/codes/deeplearn/3d/3D-CVF/second/builder/target_assigner_builder.py", line 3, in
from second.core.target_assigner import TargetAssigner
File "/home/codes/deeplearn/3d/3D-CVF/second/core/target_assigner.py", line 1, in
from second.core import box_np_ops
File "/home/codes/deeplearn/3d/3D-CVF/second/core/box_np_ops.py", line 5, in
from second.core.non_max_suppression.nms_gpu import rotate_iou_gpu_eval
File "/home/codes/deeplearn/3d/3D-CVF/second/core/non_max_suppression/init.py", line 2, in
from second.core.non_max_suppression.nms_gpu import (nms_gpu, rotate_iou_gpu,
File "/home/codes/deeplearn/3d/3D-CVF/second/core/non_max_suppression/nms_gpu.py", line 11, in
import iou3d_nms_cuda
ModuleNotFoundError: No module named 'iou3d_nms_cuda'

I use the envirment "docker pull yckimm/second:second_v1.5", and get this error.
could you fix it

@Mat8e
Copy link

Mat8e commented Aug 18, 2021

Hi,

Launching this repo requires additional installation not mentioned in the instructions.
For this issue you need to install OpenPCDet and follow the INSTALL.md instructions.

Then you need to change line 10 of second/core/non_max_suppression/nms_gpu.py with the path relative to your OpenPCDet installation : sys.path.append('/<your path to OpenPCDet parent directory>/PCDet/pcdet/ops/iou3d_nms')

@RickOnEarth
Copy link

Hi, have you faced the problem
from spconv.utils import rbbox_iou, rbbox_iou_loss ImportError: cannot import name 'rbbox_iou_loss' ?
Could you tell me which spconv version you used? I have tried all the spconv various, but none of them applys 'rbbox_iou_loss'.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants