vigra - Non-linear Diffusion and Total Variation

[ VIGRA Homepage | Function Index | Class Index | Namespaces | File List | Main Page ]

| Non-linear Diffusion and Total Variation |

## Classes
class | DiffusivityFunctor< Value >
| Diffusivity functor for non-linear diffusion. More...
## Functions
template<... >
void | anisotropicTotalVariationFilter (...)
| Performs Anisotropic Total Variation Regularization.
template<... >
void | getAnisotropy (...)
| Sets up directional data for anisotropic regularization.
template<... >
void | nonlinearDiffusion (...)
| Perform edge-preserving smoothing at the given scale.
template<... >
void | secondOrderTotalVariationFilter (...)
| Performs Anisotropic Total Variation Regularization.
template<... >
void | totalVariationFilter (...)
| Performs standard Total Variation Regularization.

## Detailed Description

Perform edge-preserving smoothing.

## Function Documentation

void vigra::nonlinearDiffusion | ( |  | ... | ) |

Perform edge-preserving smoothing at the given scale.

The algorithm solves the non-linear diffusion equation

where t is the time, x is the location vector, u( x , t) is the smoothed image at time t, and g(.) is the location dependent diffusivity. At time zero, the image u( x , 0) is simply the original image. The time is proportional to the square of the scale parameter: . The diffusion equation is solved iteratively according to the Additive Operator Splitting Scheme (AOS) from

J. Weickert: "Recursive Separable Schemes for Nonlinear Diffusion Filters", in: B. ter Haar Romeny, L. Florack, J. Koenderingk, M. Viergever (eds.): 1st Intl. Conf. on Scale-Space Theory in Computer Vision 1997, Springer LNCS 1252

`DiffusivityFunctor` implements the gradient dependent local diffusivity. It is passed as an argument to gradientBasedTransform(). The return value must be between 0 and 1 and determines the weight a pixel gets when its neighbors are smoothed. Weickert recommends the use of the diffusivity implemented by class DiffusivityFunctor. It's also possible to use other functors, for example one that always returns 1, in which case we obtain the solution to the linear diffusion equation, i.e. Gaussian convolution.

The source value type must be a linear space with internal addition, scalar multiplication, and NumericTraits defined. The value_type of the DiffusivityFunctor must be the scalar field over wich the source value type's linear space is defined.

In addition to `nonlinearDiffusion()`, there is an algorithm `nonlinearDiffusionExplicit()` which implements the Explicit Scheme described in the above article. Both algorithms have the same interface, but the explicit scheme gives slightly more accurate approximations of the diffusion process at the cost of much slower processing.

Declarations:

pass arguments explicitly:

        namespace vigra {
            template <class SrcIterator, class SrcAccessor,
                      class DestIterator, class DestAccessor,
                      class DiffusivityFunctor>
            void nonlinearDiffusion(SrcIterator sul, SrcIterator slr, SrcAccessor as,
                                    DestIterator dul, DestAccessor ad,
                                    DiffusivityFunctor const & weight, double scale);
        }

use argument objects in conjunction with Argument Object Factories :

        namespace vigra {
            template <class SrcIterator, class SrcAccessor,
                      class DestIterator, class DestAccessor,
                      class DiffusivityFunctor>
            void nonlinearDiffusion(
                      triple<SrcIterator, SrcIterator, SrcAccessor> src,
                      pair<DestIterator, DestAccessor> dest,
                      DiffusivityFunctor const & weight, double scale);
        }

Usage:

#include <vigra/nonlineardiffusion.hxx>

        FImage src(w,h), dest(w,h);
        float edge_threshold, scale;
        ...

        nonlinearDiffusion(srcImageRange(src), destImage(dest),
                           DiffusivityFunctor<float>(edge_threshold), scale);

Required Interface:

- `SrcIterator` and `DestIterator` are models of ImageIterator
- `SrcAccessor` and `DestAccessor` are models of StandardAccessor
- `SrcAccessor::value_type` is a linear space
- `DiffusivityFunctor` conforms to the requirements of gradientBasedTransform(). Its range is between 0 and 1.

- `DiffusivityFunctor::value_type` is an algebraic field

Precondition:

`scale > 0`

Examples:
  smooth.cxx.

void vigra::totalVariationFilter | ( |  | ... | ) |

Performs standard Total Variation Regularization.

The algorithm minimizes

where are the two dimensional noisy data, are the smoothed data, is the filter parameter and is the total variation semi-norm.

Declarations:

    namespace vigra {
          template <class stride1,class stride2>
          void totalVariationFilter(MultiArrayView<2,double,stride1> data,
                                    MultiArrayView<2,double,stride2> out,
                                    double alpha,
                                    int steps,
                                    double eps=0);
          void totalVariationFilter(MultiArrayView<2,double,stride1> data,
                                    MultiArrayView<2,double,stride2> weight,
                                    MultiArrayView<2,double,stride3> out,
                                    double alpha,
                                    int steps,
                                    double eps=0);
    }

totalVariationFilter() implements a primal-dual algorithm to solve (1).

Input:

data: | input data to be smoothed.
alpha: | smoothing parameter.
steps: | maximal number of iteration steps.
eps: | The algorithm stops, if the primal-dual gap is below the threshold eps.

Output:

out contains the filtered data.

In addition, a point-wise weight ( ) for the data term can be provided (overloaded function).

Usage:

#include <vigra/tv_filter.hxx>

        MultiArray<2,double> data(Shape2(width,height));  // to be initialized
        MultiArray<2,double> out(Shape2(width,height));
        MultiArray<2,double> weight(Shape2(width,height))); //optional argument in overloaded function, to be initialized if used
        int steps;        // to be initialized
        double alpha,eps; // to be initialized

        totalVariationFilter(data,out,alpha,steps,eps);

or

        totalVariationFilter(data,weight,out,alpha,steps,eps);

Examples:
  total_variation.cxx.

void vigra::getAnisotropy | ( |  | ... | ) |

Sets up directional data for anisotropic regularization.

This routine provides a two-dimensional normalized vector field , which is normal to edges in the given data, found as the eigenvector of the structure tensor belonging to the largest eigenvalue. is encoded by a scalar field of angles, i.e. .

In addition, two scalar fields and are generated from scalar parameters and , such that

at edges,
in homogeneous regions,
.

Declarations:

    namespace vigra {
    void getAnisotropy(MultiArrayView<2,double,stride1> data,
                       MultiArrayView<2,double,stride2> phi,
                       MultiArrayView<2,double,stride3> alpha,
                       MultiArrayView<2,double,stride4> beta,
                       double alpha_par,
                       double beta_par,
                       double sigma_par,
                       double rho_par,
                       double K_par);
    }

Output:

Three scalar fields phi, alpha and beta.

Input:

data: | two-dimensional scalar field.
alpha_par,beta_par: | two positive values for setting up the scalar fields alpha and beta
sigma_par: | non-negative parameter for presmoothing the data.
rho_par: | non-negative parameter for presmoothing the structure tensor.
K_par: | positive edge sensitivity parameter.

(see anisotropicTotalVariationFilter() and secondOrderTotalVariationFilter() for usage in an application).

Examples:
  total_variation.cxx.

void vigra::anisotropicTotalVariationFilter | ( |  | ... | ) |

Performs Anisotropic Total Variation Regularization.

The algorithm minimizes

where are the noisy data, are the smoothed data, is the image gradient in the sense of Total Variation and is a locally varying symmetric, positive definite 2x2 matrix.

Matrix is described by providing for each data point a normalized eigenvector (via angle ) and two eigenvalues and .

getAnisotropy() can be use to set up such data by providing a vector field normal to edges.

Declarations:

    namespace vigra {
      template <class stride1,class stride2,class stride3,class stride4,class stride5,class stride6>
      void anisotropicTotalVariationFilter(MultiArrayView<2,double,stride1> data,
                                           MultiArrayView<2,double,stride2> weight,
                                           MultiArrayView<2,double,stride3> phi,
                                           MultiArrayView<2,double,stride4> alpha,
                                           MultiArrayView<2,double,stride5> beta,
                                           MultiArrayView<2,double,stride6> out,
                                           int steps);
    }

anisotropicTotalVariationFilter() implements a primal-dual algorithm to solve (2).

Input:

data: | input data to be filtered.
steps: | iteration steps.
weight : | a point-wise weight ( ) for the data term.
phi,alpha and beta : | describe matrix , see above.

Output:

out : | contains filtered data.

Usage:

E.g. with a solution-dependent adaptivity cf. [1], by updating the matrix in an outer loop:

#include <vigra/tv_filter.hxx>

    MultiArray<2,double> data(Shape2(width,height)); //to be initialized
    MultiArray<2,double> out (Shape2(width,height));
    MultiArray<2,double> weight(Shape2(width,height));  //to be initialized
    MultiArray<2,double> phi  (Shape2(width,height));
    MultiArray<2,double> alpha(Shape2(width,height));
    MultiArray<2,double> beta (Shape2(width,height));
    double alpha0,beta0,sigma,rho,K;  //to be initialized
    int outer_steps,inner_steps;//to be initialized

    out=data; // data serves as initial value

    for (int i=0;i<outer_steps;i++){
      getAnisotropy(out,phi,alpha,beta,alpha0,beta0,sigma,rho,K);  // sets phi, alpha, beta
      anisotropicTotalVariationFilter(data,weight,phi,alpha,beta,out,inner_steps);
      }

[1] Frank Lenzen, Florian Becker, Jan Lellmann, Stefania Petra and Christoph Schnörr, A Class of Quasi-Variational Inequalities for Adaptive Image Denoising and Decomposition, Computational Optimization and Applications, Springer, 2012.

Examples:
  total_variation.cxx.

void vigra::secondOrderTotalVariationFilter | ( |  | ... | ) |

Performs Anisotropic Total Variation Regularization.

The algorithm minimizes

where are the noisy data, are the smoothed data, is the image gradient in the sense of Total Variation, is a locally varying symmetric, positive-definite 2x2 matrix and is the Frobenius norm of the Hessian of .

Matrix is described by providing for each data point a normalized eigenvector (via angle ) and two eigenvalues and . getAnisotropy() can be use to set up such data by providing a vector field normal to edges.

is the locally varying regularization parameter for second order.

Declarations:

    namespace vigra {
      template <class stride1,class stride2,...,class stride9>
      void secondOrderTotalVariationFilter(MultiArrayView<2,double,stride1> data,
                                           MultiArrayView<2,double,stride2> weight,
                                           MultiArrayView<2,double,stride3> phi,
                                           MultiArrayView<2,double,stride4> alpha,
                                           MultiArrayView<2,double,stride5> beta,
                                           MultiArrayView<2,double,stride6> gamma,
                                           MultiArrayView<2,double,stride7> xedges,
                                           MultiArrayView<2,double,stride8> yedges,
                                           MultiArrayView<2,double,stride9> out,
                                           int steps);
    }

secondOrderTotalVariationFilter() implements a primal-dual algorithm to solve (3).

Input:

data: | the input data to be filtered.
steps : | number of iteration steps.
out : | contains the filtered data.
weight : | point-wise weight ( ) for the data term.
phi,alpha,beta: | describe matrix , see above.
xedges and yedges : | binary arrays indicating the presence of horizontal (between (x,y) and (x+1,y)) and vertical edges (between (x,y) and (x,y+1)). These data are considered in the calculation of , such that finite differences across edges are artificially set to zero to avoid second order smoothing over edges.

Usage:

E.g. with a solution-dependent adaptivity (cf.[1]), by updating the matrix in an outer loop:

#include <vigra/tv_filter.hxx>

    MultiArray<2,double> data(Shape2(width,height)); //to be initialized
    MultiArray<2,double> out(Shape2(width,height));
    MultiArray<2,double> weight(Shape2(width,height))); //to be initialized
    MultiArray<2,double> phi(Shape2(width,height);
    MultiArray<2,double> alpha(Shape2(width,height);
    MultiArray<2,double> beta(Shape2(width,height));
    MultiArray<2,double> gamma(Shape2(width,height));  //to be initialized
    MultiArray<2,double> xedges(Shape2(width,height));  //to be initialized
    MultiArray<2,double> yedges(Shape2(width,height));  //to be initialized

    double alpha0,beta0,sigma,rho,K;  //to be initialized
    int outer_steps,inner_steps;//to be initialized

    out=data; // data serves as initial value

    for (int i=0;i<outer_steps;i++){

      getAnisotropy(out,phi,alpha,beta,alpha0,beta0,sigma,rho,K);  // sets phi, alpha, beta
      secondOrderTotalVariationFilter(data,weight,phi,alpha,beta,gamma,xedges,yedges,out,inner_steps);
    }

[1] Frank Lenzen, Florian Becker, Jan Lellmann, Stefania Petra and Christoph Schnörr, A Class of Quasi-Variational Inequalities for Adaptive Image Denoising and Decomposition, Computational Optimization and Applications, Springer, 2012.

Examples:
  total_variation.cxx.

© Ullrich Köthe (ullrich.koethe@iwr.uni-heidelberg.de) Heidelberg Collaboratory for Image Processing, University of Heidelberg, Germany | html generated using doxygen and Python vigra 1.9.0 (Tue Nov 6 2012)