omron-sinicx / edgeselectivefeatureweaving Goto Github PK

Hi, when i run ' python lit_corrnet3d_ESFW.py --batch_size=10 --data_dir=./trainset.h5 --test_data_dir=./testset.h5 --num_gpus 0'
the following errors occurs, do you know the reason? Thanks!

Traceback (most recent call last):
File "lit_corrnet3d_ESFW.py", line 13, in
from knn_cuda import KNN
File "/home/ch/anaconda3/envs/regtr/lib/python3.8/site-packages/knn_cuda/init.py", line 38, in
_knn = load_cpp_ext("knn")
File "/home/ch/anaconda3/envs/regtr/lib/python3.8/site-packages/knn_cuda/init.py", line 26, in load_cpp_ext
ext = load(
File "/home/ch/anaconda3/envs/regtr/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1080, in load
return _jit_compile(
File "/home/ch/anaconda3/envs/regtr/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1318, in _jit_compile
return _import_module_from_library(name, build_directory, is_python_module)
File "/home/ch/anaconda3/envs/regtr/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1701, in _import_module_from_library
module = importlib.util.module_from_spec(spec)
ImportError: /home/ch/anaconda3/envs/regtr/lib/python3.8/site-packages/knn_cuda/csrc/_ext/knn/knn.so: cannot open shared object file: No such file or directory

In the provided training data :
Keys: ['xyz1', 'xyz2'] xzy1: (1024, 3) Dataset_xyz1: , Shape: (1024, 3), Dtype: float32
& xzy2: (230000, 1024, 3) Dataset_xyz2: , Shape: (230000, 1024, 3), Dtype: float32 I think i am not being able to understand the dataset structure! why xyz1 has only 1 point cloud and xyz2 has 230k ? i would be really grateful if anyone can explain me that !