1.1 diff -r 5edfbd3e7a46 -r 1204ebf9340d PTdecode/CImg-1.3.0/examples/use_nlmeans.cpp 1.2 --- /dev/null Thu Jan 01 00:00:00 1970 +0000 1.3 +++ b/PTdecode/CImg-1.3.0/examples/use_nlmeans.cpp Mon Aug 03 14:09:20 2009 +0100 1.4 @@ -0,0 +1,133 @@ 1.5 +/* 1.6 + # 1.7 + # File : use_nlmeans.cpp 1.8 + # ( C++ source file ) 1.9 + # 1.10 + # Description : Example of use for the CImg plugin 'plugins/nlmeans.h'. 1.11 + # This file is a part of the CImg Library project. 1.12 + # ( http://cimg.sourceforge.net ) 1.13 + # 1.14 + # Copyright : Jerome Boulanger 1.15 + # ( http://www.irisa.fr/vista/Equipe/People/Jerome.Boulanger.html ) 1.16 + # 1.17 + # Benchmark : (CPU intel pentium 4 2.60GHz) compiled with cimg_debug=0. 1.18 + # patch lambda* alpha T sigma PSNR 1.19 + # 3x3 15 9x9 3.6s 20 28.22 1.20 + # 5x5 17 15x15 22.2s 20 27.91 1.21 + # 7x7 42 21x21 80.0s 20 28.68 1.22 + # 1.23 + # License : CeCILL v2.0 1.24 + # ( http://www.cecill.info/licences/Licence_CeCILL_V2-en.html ) 1.25 + # 1.26 + # This software is governed by the CeCILL license under French law and 1.27 + # abiding by the rules of distribution of free software. You can use, 1.28 + # modify and/ or redistribute the software under the terms of the CeCILL 1.29 + # license as circulated by CEA, CNRS and INRIA at the following URL 1.30 + # "http://www.cecill.info". 1.31 + # 1.32 + # As a counterpart to the access to the source code and rights to copy, 1.33 + # modify and redistribute granted by the license, users are provided only 1.34 + # with a limited warranty and the software's author, the holder of the 1.35 + # economic rights, and the successive licensors have only limited 1.36 + # liability. 1.37 + # 1.38 + # In this respect, the user's attention is drawn to the risks associated 1.39 + # with loading, using, modifying and/or developing or reproducing the 1.40 + # software by the user in light of its specific status of free software, 1.41 + # that may mean that it is complicated to manipulate, and that also 1.42 + # therefore means that it is reserved for developers and experienced 1.43 + # professionals having in-depth computer knowledge. Users are therefore 1.44 + # encouraged to load and test the software's suitability as regards their 1.45 + # requirements in conditions enabling the security of their systems and/or 1.46 + # data to be ensured and, more generally, to use and operate it in the 1.47 + # same conditions as regards security. 1.48 + # 1.49 + # The fact that you are presently reading this means that you have had 1.50 + # knowledge of the CeCILL license and that you accept its terms. 1.51 + # 1.52 +*/ 1.53 + 1.54 +#define cimg_plugin "plugins/nlmeans.h" 1.55 +#include "CImg.h" 1.56 +using namespace cimg_library; 1.57 + 1.58 +// The lines below are necessary when using a non-standard compiler as visualcpp6. 1.59 +#ifdef cimg_use_visualcpp6 1.60 +#define std 1.61 +#endif 1.62 +#ifdef min 1.63 +#undef min 1.64 +#undef max 1.65 +#endif 1.66 + 1.67 +#ifndef cimg_imagepath 1.68 +#define cimg_imagepath "img/" 1.69 +#endif 1.70 + 1.71 +int main(int argc,char **argv) { 1.72 + 1.73 + // Read command line argument s 1.74 + //----------------------------- 1.75 + cimg_usage("Non-local means denoising algorithm.\n [1] Buades, A. Coll, B. and Morel, J.: A review of image " 1.76 + "denoising algorithms, with a new one. Multiscale Modeling and Simulation: A SIAM Interdisciplinary " 1.77 + "Journal 4 (2004) 490-530 \n [2] Gasser, T. Sroka,L. Jennen Steinmetz,C. Residual variance and residual " 1.78 + "pattern nonlinear regression. Biometrika 73 (1986) 625-659 \n Build : "); 1.79 + 1.80 + // input/output and general options 1.81 + const char *file_i = cimg_option("-i",cimg_imagepath "milla.bmp","Input image"); 1.82 + const char *file_o = cimg_option("-o",(char*)NULL,"Output file"); 1.83 + const double zoom = cimg_option("-zoom",1.0,"Image magnification"); 1.84 + const double noiseg = cimg_option("-ng",0.0,"Add gauss noise before aplying the algorithm"); 1.85 + const double noiseu = cimg_option("-nu",0.0,"Add uniform noise before applying the algorithm"); 1.86 + const double noises = cimg_option("-ns",0.0,"Add salt&pepper noise before applying the algorithm"); 1.87 + const unsigned int visu = cimg_option("-visu",1,"Visualization step (0=no visualization)"); 1.88 + 1.89 + // non local means options 1.90 + const int patch_size = cimg_option("-p",1,"Half size of the patch (2p+1)x(2p+1)"); 1.91 + const float lambda = (float)cimg_option("-lambda",-1.0f,"Bandwidth as defined in [1] (-1 : automatic bandwidth)"); 1.92 + const double sigma = cimg_option("-sigma",-1,"Noise standard deviation (-1 : robust estimation)"); 1.93 + const int alpha = cimg_option("-alpha",3,"Neighborhood size (3)"); 1.94 + const int sampling = cimg_option("-sampling",1,"Sampling of the patch (1: slow, 2: fast)"); 1.95 + 1.96 + // Read image 1.97 + //------------ 1.98 + CImg<> img; 1.99 + if (file_i) { 1.100 + img = CImg<>(file_i); 1.101 + if (zoom>1) 1.102 + img.resize((int)(img.dimx()*zoom),(int)(img.dimy()*zoom),(int)(img.dimz()*zoom),-100,3); 1.103 + } else throw CImgException("You need to specify at least one input image (option -i)"); 1.104 + CImg<> original=img; 1.105 + 1.106 + // Add some noise 1.107 + //----------------- 1.108 + img.noise(noiseg,0).noise(noiseu,1).noise(noises,2); 1.109 + 1.110 + // Apply the filter 1.111 + //--------------------- 1.112 + long tic = cimg::time(); 1.113 + CImg<> dest; 1.114 + dest = img.get_nlmeans(patch_size,lambda,alpha,sigma,sampling); 1.115 + long tac = cimg::time(); 1.116 + 1.117 + // Save result 1.118 + //----------------- 1.119 + if (file_o) dest.cut(0,255.f).save(file_o); 1.120 + 1.121 + // Display (option -visu) 1.122 + //------------------- 1.123 + if (visu){ 1.124 + fprintf(stderr,"Image computed in %f s \n",(float)(tac-tic)/1000.); 1.125 + fprintf(stderr,"The pnsr is %f \n",20.*std::log10(255./std::sqrt( (dest-original).pow(2).sum()/original.size() ))); 1.126 + if (noiseg==0 && noiseu==0 && noises==0) 1.127 + CImgList<>(original,dest,((dest-original)*=2)+=128).display("Original + Restored + Estimated Noise"); 1.128 + 1.129 + else { 1.130 + CImgList<>(original,img,dest,((dest-img)*=2)+=128,((dest-original)*=2)+=128).display("Original + Noisy + Restored + Estimated Noise + Original Noise"); 1.131 + } 1.132 + } 1.133 + 1.134 + return 0; 1.135 +} 1.136 + 1.137 +