drprojects / superpoint_transformer

Official PyTorch implementation of Superpoint Transformer introduced in [ICCV'23] "Efficient 3D Semantic Segmentation with Superpoint Transformer" and SuperCluster introduced in [3DV'24 Oral] "Scalable 3D Panoptic Segmentation As Superpoint Graph Clustering"
MIT License
508 stars 65 forks source link

No module named 'frnn' #94

Closed narges-tk closed 3 months ago

narges-tk commented 3 months ago

Hi! Thanks for publishing your code! While everything went well in installation phase, now I am getting No module named 'frnn' error. Any help would be appreciated. Please find the complete error below.

ntakhtke  ❲c❳ spt  ~/DL_models/superpoint_transformer-master  1  python src/train.py experiment=semantic/narges_data Traceback (most recent call last): File "src/train.py", line 49, in from src import utils File "/home/ntakhtke/DL_models/superpoint_transformer-master/src/init.py", line 2, in import src.data File "/home/ntakhtke/DL_models/superpoint_transformer-master/src/data/init.py", line 1, in from .csr import File "/home/ntakhtke/DL_models/superpoint_transformer-master/src/data/csr.py", line 5, in from src.utils import tensor_idx, is_sorted, indices_to_pointers, \ File "/home/ntakhtke/DL_models/superpoint_transformer-master/src/utils/init.py", line 12, in from .neighbors import File "/home/ntakhtke/DL_models/superpoint_transformer-master/src/utils/neighbors.py", line 3, in from src.dependencies.FRNN import frnn File "/home/ntakhtke/DL_models/superpoint_transformer-master/src/dependencies/FRNN/frnn/init.py", line 2, in from .frnn import frnn_grid_points, frnn_gather, frnn_bf_points, _C File "/home/ntakhtke/DL_models/superpoint_transformer-master/src/dependencies/FRNN/frnn/frnn.py", line 7, in from frnn import _C ModuleNotFoundError: No module named 'frnn'

drprojects commented 3 months ago

Hi @narges-tk, thanks for your interest in this project.

This error comes from the fact that the dependency FRNN was not properly installed when running install.sh.

This may be related to #77, please check this first.

PS: if you ❤️ or use this project, don't forget to give it a ⭐, it means a lot to us !

narges-tk commented 3 months ago

Thanks for your prompt reply. I resolved it by setting CUDA_HOME as follows: export CUDA_HOME=/usr/local/cuda-12.1