Tested with Python3.8 and CUDA 11.3.
git clone https://github.com/TRAILab/STSLidR.git
conda create -n stslidr python=3.8
conda activate stslidr
pip install torch==1.12.0 torchvision==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu113
git clone --recursive "https://github.com/NVIDIA/MinkowskiEngine"` \
cd MinkowskiEngine
python setup.py install --force_cuda --blas=openblas
cd ..
pip install .[torch]
pip install .
Add symbolic link to nuScenes and SemanticKITTI datasets under datasets folder
ln -s /mnt/disks/wd_ssd/nuscences_dataset/data/sets/nuscenes/ ./datasets/nuscenes
ln -s /mnt/disks/wd_ssd/semantic_kitti/dataset/ ./datasets/kitti
Generate SLIC superpixels in superpixels/nuscenes/superpixels_slic directory
python superpixel_segmenter.py