yuanhaogong / curvaturefilter Goto Github PK

These curvature filters are developed by Yuanhao Gong during his PhD. MC filter and TV filter are exactly the same as described in the paper. But the GC filter is slightly modified. Please cite following papers if you use curvature filter in your work. Thank you!

📚 The Paper. The general theory is in Chapter Six of PhD thesis (downloaded 17,000+)

📕 Presentation of Gaussian Curvature Filter: LinkedIn, Dropbox or Baidu.

📘 Poster of Bernstein Filter can be found here.

🎁 a short introduction in Chinese (中文): Zhihu(Editors' Choice) or this Zhihu

🏆 source code in C++ and Java can also be found from MOSAIC group

🔔 The kernels summary and one example how to get the kernel can be found here.

📧 [email protected] or join the Curvature Filter Forum

@ARTICLE{gong:cf, 
    author={Yuanhao Gong and Ivo F. Sbalzarini}, 
    journal={IEEE Transactions on Image Processing}, 
    title={Curvature filters efficiently reduce certain variational energies}, 
    year={2017}, 
    volume={26}, 
    number={4}, 
    pages={1786-1798}, 
    doi={10.1109/TIP.2017.2658954}, 
    ISSN={1057-7149}, 
    month={April},}

@phdthesis{gong:phd, 
    title={Spectrally regularized surfaces}, 
    author={Gong, Yuanhao}, 
    year={2015}, 
    school={ETH Zurich, Nr. 22616},
    note={http://dx.doi.org/10.3929/ethz-a-010438292}}
	
@article{gong:gc,
    Author = {Yuanhao Gong and Ivo F. Sbalzarini},
    Journal = {Intl. Conf. Image Proc. (ICIP)},
    Month = {September},
    Pages = {534--538},
    Title = {Local weighted {G}aussian curvature for image processing},
    Year = {2013}}

Traditional solvers, such as gradient descent or Euler Lagrange Euqation, start at the total energy and use diffusion scheme to carry out the minimization. When the initial condition is the original image, the data fitting energy always increases while the regularization energy always reduces during the optimization, as illustrated in the below figure. Thus, regularization energy must be the dominant part since the total energy has to decrease.

Therefore, Curvature filters focus on minimizing the regularization term, whose minimizers are already known. For example, if the regularization is Gaussian curvature, the developable surfaces minimize this energy. Therefore, in curvature filter, developable surfaces are used to approximate the data. As long as the decreased amount in the regularization part is larger than the increased amount in the data fitting energy, the total energy is reduced.

Theoretical	Practical
Generality: handle arbitrary data fitting term (BlackBox)	Efficient: three or four order of magnitude faster than traditional solvers
Convergence: theoretically guaranteed	Implementation: 40 lines in Matlab and 100 lines in C++

Running time with 10 iterations on 512X512 Lena image. Matlab version is R2015a and GCC version is 5.1. All tests are on a Thinkpad T410 with i7-620M core CPU (2.6GHz). We take the time for 100 iterations and divide it by 10. On average, curvature filters take 1 millisecond per iteration.

On my new taptop(Thinkpad T470p, NVIDIA GeForce 940MX, 384 CUDA cores), GPU version of MC filter can achieve 2500 MPixels/Second with shared memory and single precision.

On the TITAN Xp card, MC filter can achieve 33.2 Giga Pixels/Second with shared memory and single precision. On the Tesla K40c card (2880 cores), MC filter can achieve 8090 MPixels/Second with shared memory and single precision.

The noise free test image can be downloaded here

GC = Gaussian Curvature, MC = Mean Curvature, TV = Total Variation

We show three lines' profiles during minimizing a mean curvature regularized model (MC filter used).

from left to right: original reference image, distorted source image, registered results by TV filter, MC filter and GC filter.

original mesh (left) and processed mesh (right), the energy profile is shown in the middle.

1. Why dual mesh (DM) structure is needed?

There are two reasons. First, these four sets guarantee the convergence. Second, we can use the updated neighbors for current position. Therefore, it is more computational efficient.

2. What is the difference between these three filters?

In general, GC filter is better in preserving details, compared with the other two. And TV filter is better in removing noise as well as details. MC filter is between these two.

These three filters are correspond to three types of variational models. User should decide which prior is to be assumed about the ground truth.

3. What is the difference between split and nosplit scheme?

In general, splitting the image into four sets and looping on them is computational faster. However, in some cases like deconvolution, we need to merge the four sets after every iteration. So, it is better do nosplit scheme.

These two lead to exactly the same result. The split code is just more cache friendly.