6DGS项目使用教程

6DGS项目使用教程

6dgs Code of the paper: 6DGS: 6D Pose Estimation from a Single Image and a 3D Gaussian Splatting Model 项目地址: https://gitcode.com/gh_mirrors/6d/6dgs

1. 项目目录结构及介绍

6DGS项目是一个基于单张图像进行6D姿态估计的开源项目。以下是项目的目录结构及文件介绍：

6dgs/
├── .idea/                   # PyCharm项目配置文件
├── arguments/               # 存储训练和测试时使用的参数文件
├── cfg_grammar/             # 存储配置文件语法相关的文件
├── gaussian_renderer/       # 3D高斯渲染器相关代码
├── lpipsPyTorch/            # 用于学习感知图像相似度的PyTorch库
├── pose_estimation/         # 姿态估计相关代码
├── scene/                   # 场景相关的处理代码
├── static/                  # 静态文件目录
├── submodules/              # 存储子模块的目录
├── tools/                   # 包含训练和测试脚本等工具
├── utils/                   # 存储一些工具类和函数的目录
├── .gitignore               # 指定git忽略的文件和目录
├── .gitmodules              # 包含子模块信息
├── LICENSE.md               # 项目许可证文件
├── README.md                # 项目说明文件
├── convert.py               # 用于数据转换的脚本
├── environment.yml          # Conda环境配置文件
├── full_eval.py             # 完整评估脚本
├── metrics.py               # 评估指标相关代码
├── pretrain_eval_attention.py # 预训练和评估注意力模型的脚本
├── render.py                # 渲染相关脚本
├── train.py                 # 训练模型的脚本

2. 项目的启动文件介绍

项目的启动主要依赖于train.py和pretrain_eval_attention.py这两个脚本。

• train.py：此文件是用于训练基础3DGS模型的脚本。通过指定数据集的位置，可以使用此脚本来训练模型。

• pretrain_eval_attention.py：此文件用于姿态估计的预训练和评估。根据不同的数据类型（例如Mip-NeRF 360或Tanks Temple），可以调整参数进行训练和测试。

3. 项目的配置文件介绍

项目的配置主要通过environment.yml和参数文件进行。

• environment.yml：这是一个Conda环境配置文件，用于创建一个隔离的环境，其中包含了项目运行所需的依赖库。

• 参数文件：在arguments/目录中，包含了多个参数文件，这些文件用于在训练或测试时指定模型的参数，如数据集位置、模型超参数等。这些文件通常以.yml或.txt格式存在，可以在运行train.py或pretrain_eval_attention.py时通过命令行参数进行指定。

6dgs Code of the paper: 6DGS: 6D Pose Estimation from a Single Image and a 3D Gaussian Splatting Model 项目地址: https://gitcode.com/gh_mirrors/6d/6dgs