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