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}
}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