Software Open Access

# An efficient C code for diffusion tensor denoising with total generalised variation

Tuomo Valkonen

Written 2012-2014 by Tuomo Valkonen tuomov@iki.fi, University of Graz & University of Cambridge.

This repository includes code for TGV^2 and TV denoising and reconstruction from sparse k-space data. For problems with linear operators, the Chambolle-Pock [3] modified primal dual hybrid gradient method (PDHGM) is used. For problems with non-linear, the non-linear PHDHGM of [5] and the Gauss-Newton method are supported. Optional Bregman iterations may be applied to many problems. The code is parallelised using OpenMP. There is also OpenACC GPU support, but at the moment it is not actively maintained.

The included programs and cases covered are the following.

• denoise: basic colour or grayscale image denoising.

• tensorreg: Diffusion tensor denoising per [2]. This solves the optionally positivity-constrained problem

min_{u >= 0} FID(f-Au) + REG(u)


for tensor fields u and f. For u, tensors of order 0 (scalar functions), 1 (vector fields), and 2 are supported. A is a pointwise operator, and FID either the L1 fidelity or the squared L2 fidelity. The regulariser REG is one of TGV^2, TD (TGV^1), or TV.

Small sub-problems in the positivity-constrained case with A!=I are solved with the primal-dual interior point method of [4].

• ksreg: Sparse reconstruction from K-space data. This solves the problem

min_u |f-SFu|^2 + REG(u)

for complex image u, and sparse K-space data f. Here S is the sparse sampling operator and F the (discrete) Fourier transform.

• ksreg-nl: The same as ksreg, but with individual regularisation of the phase and magnitude of u, and non-linear reconstruction of u from these.

Contents

doc/            Documentation, include a user guide to the
tensorreg program, and quick developer
introduction to the C code.

examples/       Multiple use-case examples, and miscellaneous
Matlab code.

src/            Source code.

matlab/         Auxiliary Matlab code.
(Plotting, NIFTI import, ...)

mex/            Auxiliary Mex routines for Matlab.
Necessary for the plotting routine!


Dependencies

GNU make        http://www.gnu.org/

Other variants of make may not work.

pgcc            http://www.pgroup.com/

Or any other OpenACC compiler for the
GPU implementation.

FFTW3           http://www.fftw.org/

For Fast Fourier Transform.

NFFT3           http://www-user.tu-chemnitz.de/~potts/nfft/

Non-uniform Fast Fourier Transform.


Compilation

To compile the code, follow the following steps

1. Download install the libtu utility library available from this same Zenodo page.

2. Copy local.mk.in as local.mk

3. Edit local.mk to suit your system and choices.

4. Run make

Third-party code

This package includes NIFTI code by Jimmy Shen (matlab/NIFTI_code), and Rice distribution code by Ged Ridgway (matlab/rician).

This work is published under GNU GPLv3. See the attached LICENSE.txt.

References

1. K. Bredies and K. Kunisch and T. Pock, Total generalised variation, SIAM Journal on Imaging Sciences 3 (2011), 492--526. http://dx.doi.org/10.1137/090769521

2. T. Valkonen, K. Bredies, and F. Knoll, Total generalised variation in diffusion tensor imaging, SIAM Journal on Imaging Sciences 6 (2013), 487–525. http://dx.doi.org/10.1137/120867172, https://tuomov.iki.fi/mathematics/dtireg.pdf>

3. A. Chambolle and T. Pock, A first-order primal-dual algorithm for convex problems with applications to imaging, Journal of Mathematical Imaging and Vision 40 (2011), 120-145. http://dx.doi.org/10.1007/s10851-010-0251-1

4. T. Valkonen, A method for weighted projections to the positive definite cone, Optimization 64 (2015), 2253–2275. https://tuomov.iki.fi/mathematics/scaleproj.pdf, https://dx.doi.org/10.1080/02331934.2014.929680

5. T. Valkonen, A primal-dual hybrid gradient method for non-linear operators with applications to MRI (2013), Inverse Problems 30 (2014),055012, http://dx.doi.org/10.1088/0266-5611/30/5/055012, http://iki.fi/tuomov/mathematics/nl-pdhgm.pdf

Files (1.2 MB)
Name Size
dticode.zip
md5:d5f6dc587270744ab2b3d77ed2612d90
1.0 MB
libtu-3.zip
md5:fc63b9ce28d6882095d0e5fd62fee758
116.2 kB
• Valkonen, Bredies and Knoll, Total generalised variation in diffusion tensor imaging, SIAM Journal on Imaging Sciences 6 (2013), 487–525, doi:10.1137/120867172

• Valkonen and Liebmann, GPU-accelerated regularisation of large diffusion tensor volumes, Computing 95 (2013), 771–784, doi:10.1007/s00607-012-0277-x, Special issue for ESCO 2012, Pilsen, Czech Republic

76
23
views
All versions This version
Views 7676
Data volume 15.7 MB15.7 MB
Unique views 6363