Unsupervised Learning of the Total Variation Flow

This repository is the official implementation of Unsupervised Learning of the Total Variation Flow by Tamara G. Grossmann, Sören Dittmer, Yury Korolev and Carola-Bibiane Schönlieb.

If you use this code, please cite:

@article{grossmann2022unsupervised,
  title={Unsupervised Learning of the Total Variation Flow},
  author={Grossmann, Tamara G and Dittmer, S{\"o}ren and Korolev, Yury and Sch{\"o}nlieb, Carola-Bibiane},
  journal={arXiv preprint arXiv:2206.04406},
  year={2022}
}

Requirements

The code was written in Pytorch version 1.12.1 for CUDA 11.4. Following packages are required to smoothly run the code:

• pytorch
• torchvision
• numpy
• scikit-image