ITKTotalVariation

.. image:: https://github.com/InsightSoftwareConsortium/ITKTotalVariation/workflows/Build,%20test,%20package/badge.svg

.. image:: https://img.shields.io/pypi/v/itk-totalvariation.svg :target: https://pypi.python.org/pypi/itk-totalvariation :alt: PyPI

.. image:: https://img.shields.io/badge/License-Apache%202.0-blue.svg :target: https://github.com/InsightSoftwareConsortium/ITKTotalVariation/blob/master/LICENSE :alt: License

Overview

TotalVariation algorithms. itkProxTVImageFilter provides a wrap for the external project proxTV: https://github.com/albarji/proxTV

ITK provides wrapping for Windows, macOS and Linux:

.. code-block:: python

import itk Dimension = 3 I = itk.Image[itk.F, Dimension] reader = itk.ImageFileReader[I].New(FileName=file_name) reader.Update() image = reader.GetOutput() tv = itk.ProxTVImageFilter[I, I].New() tv.SetInput(image) tv.SetMaximumNumberOfIterations(10) tv.SetWeights(100) tv.Update() output_file_name="/tmp/tvout.nrrd" itk.imwrite(tv.GetOutput(), output_file_name)

Examples of 3D image denoised

Original:

.. image:: https://user-images.githubusercontent.com/3021667/55841022-95524a00-5afb-11e9-92f1-3743c4ddbf4c.png

TotalVariation denoised:

.. image:: https://user-images.githubusercontent.com/3021667/55841021-94b9b380-5afb-11e9-9961-37072274bd68.png

Original:

.. image:: https://user-images.githubusercontent.com/3021667/38002434-9fc35d32-3232-11e8-8bfc-a7d9ce6888c7.png

Denoised:

.. image:: https://user-images.githubusercontent.com/3021667/38002451-b87bed62-3232-11e8-814b-50e8fb0f79cb.png