OctNet 项目使用教程

OctNet 项目使用教程

octnet OctNet: Learning Deep 3D Representations at High Resolutions 项目地址: https://gitcode.com/gh_mirrors/oc/octnet

1. 项目目录结构及介绍

OctNet 是一个使用高效空间划分结构（即八叉树）来减少3D卷积神经网络内存和计算需求的开源项目，这使得在高分辨率下进行深度学习成为可能。项目的目录结构如下：

• core：包含八叉树数据结构的核心代码，CPU上的网络操作代码以及一些用于创建测试对象的代码。
• core_gpu：GPU（CUDA）上的网络操作代码。
• create：用于预处理3D数据（点云、网格、密集体积）并将其转换为八叉树结构的代码。
• geometry：主要用于 create 包中的一些简单几何例程。
• py：一个小的Python包装器，用于 create 包和 core 函数的一些操作。
• th：一个功能完整的torch包装器，用于所有网络操作。
• example：包含一个示例，用于创建数据并在ModelNet10上训练网络。
• doc：存放项目文档。
• build_all.sh 和 build_cpu.sh：用于自动构建项目的bash脚本。

2. 项目的启动文件介绍

项目的启动主要依赖于构建脚本。以下是两个主要的启动文件：

• build_cpu.sh：构建所有CPU代码的脚本。在 core 目录下运行此脚本将编译CPU版本的八叉树结构和网络操作。

  cd core
  mkdir build
  cd build
  cmake ..
  make -j
  

• build_all.sh：除了构建CPU代码，还会构建GPU网络函数和GPU上torch的包装器代码。

  cd core
  python setup.py build_ext --inplace
  

3. 项目的配置文件介绍

OctNet 的配置主要通过CMake来进行。在构建之前，你需要在 core 目录下创建一个 build 文件夹，并运行 cmake 命令来配置项目。以下是一个基本的配置步骤：

• 创建一个构建目录：

  mkdir build
  cd build
  

• 运行CMake来配置项目：

  cmake ..
  

• 使用 make 命令构建项目：

  make -j
  

在CMake配置过程中，你可以通过设置不同的参数来调整构建选项，例如启用或禁用OpenMP等。

请根据你的需要和系统环境进行适当的配置和构建。在构建完成后，你可以通过 example 目录中的示例来开始训练和测试网络。

octnet OctNet: Learning Deep 3D Representations at High Resolutions 项目地址: https://gitcode.com/gh_mirrors/oc/octnet