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 感知損失。