deepinv/deepinv
DeepInverse: a PyTorch library for solving imaging inverse problems using deep learning
解決的問題
DeepInverse 致力於解決成像逆問題。它提供了一個模組化框架，用於加速各種成像領域的深度學習研究，提高可複現性並降低新從業者進入門檻。
工作原理
該函式庫基於 PyTorch 構建，提供以下核心組件：
- 大量的預定義成像算子框架。
- 最先進的深度神經網路，包括預訓練的重建模型與去噪器。
- 用於即插即用式修復、優化與 unfolded 架構的框架。
- 針對逆問題的專用訓練損失函數。
- 用於不確定性量化的採樣演算法與擴散模型。
- 用於構建逆問題專用資料集的專用框架。
適用對象
專為成像領域的研究人員與從業者設計，他們需要一種標準化、模組化的方式來實現與測試用於逆問題的深度學習演算法。
亮點
- PyTorch 生態系統的一部分
- 支援即插即用的修復與 unfolded 架構
- 包含用於不確定性量化的擴散模型
- 提供構建專用資料集的框架