cann-recipes-spatial-intelligence
🚀Latest News
- [2025/12] 3DGS算法模型在昇腾Atlas A2系列上已支持训练和推理,代码已开源。
- [2025/11] Hunyuan3D模型在昇腾Atlas A2系列上已支持推理,代码已开源。
- [2025/11] VGGT模型在昇腾Atlas A2系列上已支持推理,代码已开源。
🎉概述
cann-recipes-spatial-intelligence仓库旨在针对空间智能业务中的典型模型、加速算法,提供基于CANN平台的优化样例,方便开发者简单、快速、高效地基于CANN平台使用空间智能模型。
✨样例列表
| 实践 | 简介 |
|---|---|
| VGGT | 基于VGGT开源模型,完成其在Atlas A2上的推理适配,并提供其在相机位姿估计、点云重建、深度估计三个任务上的精度评测脚本。 |
| Hunyuan3D | 基于Hunyuan3D开源模型,完成其在Atlas A2上的推理适配,并通过使能融合算子、图模式、多线程并行光栅化等优化手段,实现了较低的推理时延。 |
| 3DGS | 基于3DGS开源加速库,完成其在Atlas A2上的训练和推理适配,并基于NPU特性进行Alpha-blending、视锥剔除、负载均衡、Precise Intersection等优化,实现较低的训推耗时。本样例提供全流程的3DGS训推脚本。 |
📖目录结构说明
├── docs # 文档目录
| └── models # 模型文档目录
| ├── Hunyuan3D # Hunyuan3D相关文档
| ├── vggt # VGGT相关文档
| └── ...
| └── algorithms # 算法文档目录
| ├── gaussian_splatting # 3DGS相关文档
| └── ...
├── models # 模型脚本目录
| ├── Hunyuan3D # Hunyuan3D的模型脚本及执行配置
| ├── vggt # vggt的模型脚本及执行配置
│ └── ...
├── algorithms # 算法脚本目录
│ ├─ gaussian_splatting # 3DGS算法脚本及执行配置
│ └── ...
├── ops # 自定义融合算子目录
│ ├─ ascendc # 昇腾平台(3DGS)融合算子目录
│ └── ...
└── contrib # 社区用户贡献的模型与文档目录
| ├── README.md
│ └── ...
└── CONTRIBUTION.md
└── DISCLAIMER.md
└── LICENSE
└── README.md
└── ...