单个代码文件就可以运行randlanet 分割网络,输入可以是 bin 或者 PCD 文件,最后还有一个显示模块把显示结果加上了
# coding:utf-8
import os
import open3d.ml as _ml3d
import open3d.ml.torch as ml3d
'''
pip install open3d -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
pip install tensorboard -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
pip install torch==1.13.1+cpu torchvision==0.14.1+cpu torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cpu
'''
import numpy as np
import open3d as o3d
# Class colors, RGB values as ints for easy reading
COLOR_MAP = {
0: (0, 0, 0),
1: (245, 150, 100),