# pointnet-mrf

**Repository Path**: fbx-research/pointnet-mrf

## Basic Information

- **Project Name**: pointnet-mrf
- **Description**: 融合马尔可夫随机场和PointNet++的点云分割模型
- **Primary Language**: Python
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 1
- **Created**: 2021-04-26
- **Last Updated**: 2024-03-25

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# PointNet++ — MRF

+ 基于PointNet++与马尔可夫随机场的点云语义分割模型

> 模型环境

+ PointNet++ 模型，模型地址：[模型](https://github.com/yanx27/Pointnet2_pytorch) 论文地址：[论文](https://arxiv.org/abs/1706.02413)

+ 马尔可夫随机场（MRF）

+ pytorch

+ torchvision

+ torchaudio

+ cudatoolkit

> 环境配置

>> conda安装

```bash
conda env create -f conda-gpu.yml
conda activate pointnet-mrf-gpu
```

>> pip安装

```bash
pip install -r requirements.txt
```


> 文档架构

```
|-- POINTNET-MRF
    |-- datasets 数据集存放目录
    |-- models 模型存放目录
    |-- running 运行目录
        |-- [类别名称]
            |-- checkpoints 检查点目录
            |-- logs 日志目录
    |-- utils 工具集目录
        |-- data_utils 数据处理工具
    |-- config.py 配置文件
    |-- train.py 训练入口
    |-- train_v2.py 训练入口2.0
    |-- train_v3.py 训练入口3.0 支持单场景训练
    |-- eval.py 测试入口
    |-- eval_v3.py 测试入口3.0 支持单场景测试
```


> 使用方法

+ 1. 训练数据处理

    + 进入 utils/data_utils 文件夹

    + 打开 s3dis_txt2npy_util.py 文件，修改路径 DATA_PATH = os.path.dirname('你的S3DIS真实目录')

    + 执行 python s3dis2npy.py

    打开config.py文件，设置模型，GPU等训练参数

+ 2. 配置参数

    打开config.py文件，设置模型，GPU等训练参数

+ 3. 开始训练

    ```
    # 开始训练
    python train.py
    ```

> 模型表现

> 分类类别

ceiling 天花板
floor 地板
wall 墙体
beam 梁
column 柱
window 窗户
door 门
table 桌子
chair 椅子
sofa 沙发
bookcase 书柜
board 板
clutter 杂物