显著性区域检测代码.docxVIP

// Saliency.h#if !defined(_SALIENCY_H_INCLUDED_) #define _SALIENCY_H_INCLUDED_ #include vector #include cfloat using namespace std; class Saliency{ public: Saliency(); virtual ~Saliency(); public: void GetSaliencyMap( const vectorunsigned int inputimg,//INPUT: ARGB buffer in row-major order const int width, const int height, vectordouble salmap,//OUTPUT: Floating point buffer in row-major order const bool normalizeflag = true);//false if normalization is not needed private: void RGB2LAB( const vectorunsigned int ubuff, vectordouble lvec, vectordouble avec, vectordouble bvec); void GaussianSmooth( const vectordouble inputImg, const int width, const int height, const vectordouble kernel, vectordouble smoothImg); //============================================================================== /// Normalize //============================================================================== void Normalize( const vectordouble input, const int width, const int height, vectordouble output, const int normrange = 255) { double maxval(0); double minval(DBL_MAX); {int i(0); for( int y = 0; y height; y++ ) { for( int x = 0; x width; x++ ) { if( maxval input[i] ) maxval = input[i]; if( minval input[i] ) minval = input[i]; i++; } }} double range = maxval-minval; if( 0 == range ) range = 1; int i(0); output.clear(); output.resize(width*height); for( int y = 0; y height; y++ ) { for( int x = 0; x width; x++ ) { output[i] = ((normrange*(input[i]-minval))/range); i++; } } } }; #endif // !defined(_SALIENCY_H_INCLUDED_) // Saliency.cpp: implementation of the Saliency class. // ////////////////////////////////////////////////////////////////////// //=========================================================================== // Copyright (c) 2009 Radhakrishna Achanta [EPFL]//=========================================================================== #