单个代码文件就可以运行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),
- 2: (24