Skip to content

An implementation of Denoising Variational AutoEncoder with Topological loss

Notifications You must be signed in to change notification settings

yuki3-18/Topological-DVAE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Topological-DVAE

An implementation of Denoising Variational AutoEncoder with Topological loss

Requirements

1. PyTorch
2. SimpleITK
3. tqdm
4. visdom
5. TopologyLayer
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

About

An implementation of Denoising Variational AutoEncoder with Topological loss

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published