ashawkey/kiuikit

解决的问题

Kiuikit 是一个专注于 3D 视觉的专用工具包，旨在简化计算机视觉中的常见任务。它提供了一系列用于处理 3D 网格、视频/图像处理以及通常需要从零开始实现的神经网络组件的实用工具。

工作原理

该工具包既作为命令行界面（CLI）也作为 Python 库运行。CLI 提供了 3D 网格查看器、超分辨率和系统/作业管理工具。Python 库则提供 torch 原生 3D 网格处理、向量与四元数数学、相机工具，以及 DiT 和 UNet 等预构建的 PyTorch 神经网络模块。

适用人群

适用于从事计算机视觉和 3D 重建的研究人员和开发者，他们需要高效的数据 I/O、几何数学工具以及构建 AI 模型的便捷手段。

主要亮点

• 3D 视觉工具：包含 GUI 3D 网格查看器（kire）、torch 原生网格加载/导出功能以及四元数数学。
• 神经网络模块：提供独立的 PyTorch 模块，包括 MLP、DiT、UNet（2D/3D）、VAE 和流匹配（flow matching）。
• 媒体 I/O：支持 float/HDR 格式，兼容 numpy/torch 的高性能视频与图像读写功能。
• 开发者工具：包含 CUDA 同步计时器、基于 CUDA 后端的哈希网格编码，以及 LPIPS 感知损失。