用户画像>> Topological-DVAE>> 返回
项目作者： yuki3-18

项目描述 ：
An implementation of Denoising Variational AutoEncoder with Topological loss
高级语言： Python
项目地址: git://github.com/yuki3-18/Topological-DVAE.git
创建时间： 2020-02-04T19:11:29Z
项目社区：https://github.com/yuki3-18/Topological-DVAE
开源协议：
下载


Topological-DVAE

An implementation of Denoising Variational AutoEncoder with Topological loss

Requirements

  1. 1. PyTorch
  2. 2. SimpleITK
  3. 3. tqdm
  4. 4. visdom
  5. 5. TopologyLayer
  6. 6. Gudhi

Usage

Training Model: main.py\
Evaluation: eval.py\
Check PH: PH.py

Reference

[1] Kingma, D. P., & Welling, M. (2013). “Auto-Encoding Variational Bayes”, (Ml), 1–14. https://doi.org/10.1051/0004-6361/201527329

[2] James R. Clough, et al. (2019). “A Topological Loss Function for Deep-Learning based Image Segmentation using Persistent Homology”. https://arxiv.org/abs/1910.01877

[3] https://github.com/pytorch/examples/tree/master/vae

[4] https://github.com/JamesClough/topograd

[5] https://github.com/bruel-gabrielsson/TopologyLayer