SLIC超像素分割算法和目前超像素算法的比较代码实现

Resource.h//{{NO_DEPENDENCIES}}// Microsoft Visual C++ generated include file.// Used by SLICSuperpixels.rc//#define IDD_SLICSUPERPIXELS_DIALOG 102#define IDR_MAINFRAME 128#define IDC_BUTTON_CREATESUPERPIXELS 1000#define IDC_EDIT_SPCOUNT 1001#define IDC_EDIT2 1002#define IDC_EDIT_COMPACTNESS 1002// Next default values for new objects//#ifdef APSTUDIO_INVOKED#ifndef APSTUDIO_READONLY_SYMBOLS#define _APS_NEXT_RESOURCE_VALUE 129#define _APS_NEXT_COMMAND_VALUE 32771#define _APS_NEXT_CONTROL_VALUE 1003#define _APS_NEXT_SYMED_VALUE 101#endif#endifStdafx.h// stdafx.h : include file for standard system include files,// or project specific include files that are used frequently,// but are changed infrequently#pragma once#ifndef _SECURE_ATL#define _SECURE_ATL 1#endif#ifndef VC_EXTRALEAN#define VC_EXTRALEAN // Exclude rarely-used stuff from Windows headers #endif#include"targetver.h"#define_ATL_CSTRING_EXPLICIT_CONSTRUCTORS // some CString constructors will be explicit// turns off MFC's hiding of some common and often safely ignored warning messages#define _AFX_ALL_WARNINGS#include<afxwin.h>// MFC core and standard components#include<afxext.h>// MFC extensions#ifndef _AFX_NO_OLE_SUPPORT#include<afxdtctl.h>// MFC support for Internet Explorer 4 Common Controls#endif#ifndef _AFX_NO_AFXCMN_SUPPORT#include<afxcmn.h>// MFC support for Windows Common Controls#endif// _AFX_NO_AFXCMN_SUPPORT#include<afxcontrolbars.h>// MFC support for ribbons and control barsSLIC.h// SLIC.h: interface for the SLIC class.//===========================================================================// This code implements the superpixel method described in://// Radhakrishna Achanta, Appu Shaji, Kevin Smith, Aurelien Lucchi, Pascal Fua, and Sabine Susstrunk,// "SLIC Superpixels",// EPFL Technical Report no. 149300, June 2010.//===========================================================================// Copyright (c) 2012 Radhakrishna Achanta [EPFL]. All rights reserved.//===========================================================================//////////////////////////////////////////////////////////////////////#if !defined(_SLIC_H_INCLUDED_)#define _SLIC_H_INCLUDED_#include<vector>#include<string>#include<algorithm>using namespace std;class SLIC{public:SLIC();virtual ~SLIC();//============================================================================// Superpixel segmentation for a given step size (superpixel size ~= step*step)//============================================================================ void DoSuperpixelSegmentation_ForGivenSuperpixelSize(const unsigned int* ubuff,//Each 32 bit unsigned int contains ARGB pixel values.const int width,const int height,int*& klabels,int& numlabels,const int& superpixelsize,const double& compactness);//============================================================================// Superpixel segmentation for a given number of superpixels//============================================================================ void DoSuperpixelSegmentation_ForGivenNumberOfSuperpixels(const unsigned int* ubuff,const int width,const int height,int*& klabels,int& numlabels,const int& K,//required number of superpixelsconst double& compactness);//10-20 is a good value for CIELAB space//============================================================================// Supervoxel segmentation for a given step size (supervoxel size ~= step*step*step) //============================================================================void DoSupervoxelSegmentation(unsigned int**& ubuffvec,const int& width,const int& height,const int& depth,int**& klabels,int& numlabels,const int& supervoxelsize,const double& compactness);//============================================================================ // Save superpixel labels in a text file in raster scan order//============================================================================ void SaveSuperpixelLabels(const int*& labels,const int& width,const int& height,const string& filename,const string& path);//============================================================================ // Save supervoxel labels in a text file in raster scan, depth order//============================================================================ void SaveSupervoxelLabels(const int**& labels,const int& width,const int& height,const int& depth,const string& filename,const string& path);//============================================================================ // Function to draw boundaries around superpixels of a given 'color'.// Can also be used to draw boundaries around supervoxels, i.e layer by layer.//============================================================================ void DrawContoursAroundSegments(unsigned int*& segmentedImage,int*& labels,const int& width,const int& height,const unsigned int& color );private://============================================================================ // The main SLIC algorithm for generating superpixels//============================================================================void PerformSuperpixelSLIC(vector<double>& kseedsl,vector<double>& kseedsa,vector<double>& kseedsb,vector<double>& kseedsx,vector<double>& kseedsy,int*& klabels,const int& STEP,const vector<double>& edgemag,const double& m = 10.0);//============================================================================ // The main SLIC algorithm for generating supervoxels//============================================================================ void PerformSupervoxelSLIC(vector<double>& kseedsl,vector<double>& kseedsa,vector<double>& kseedsb,vector<double>& kseedsx,vector<double>& kseedsy,vector<double>& kseedsz,int**& klabels,const int& STEP,const double& compactness);//============================================================================ // Pick seeds for superpixels when step size of superpixels is given.//============================================================================ void GetLABXYSeeds_ForGivenStepSize(vector<double>& kseedsl,vector<double>& kseedsa,vector<double>& kseedsb,vector<double>& kseedsx,vector<double>& kseedsy,const int& STEP,const bool& perturbseeds,const vector<double>& edgemag);//============================================================================ // Pick seeds for supervoxels//============================================================================ void GetKValues_LABXYZ(vector<double>& kseedsl,vector<double>& kseedsa,vector<double>& kseedsb,vector<double>& kseedsx,vector<double>& kseedsy,vector<double>& kseedsz,const int& STEP);//============================================================================ // Move the superpixel seeds to low gradient positions to avoid putting seeds // at region boundaries.//============================================================================ void PerturbSeeds(vector<double>& kseedsl,vector<double>& kseedsa,vector<double>& kseedsb,vector<double>& kseedsx,vector<double>& kseedsy,const vector<double>& edges);//============================================================================ // Detect color edges, to help PerturbSeeds()//============================================================================ void DetectLabEdges(const double* lvec,const double* avec,const double* bvec,const int& width,const int& height,vector<double>& edges);//============================================================================ // sRGB to XYZ conversion; helper for RGB2LAB()//============================================================================ void RGB2XYZ(const int& sR,const int& sG,const int& sB,double& X,double& Y,double& Z);//============================================================================ // sRGB to CIELAB conversion (uses RGB2XYZ function)//============================================================================ void RGB2LAB(const int& sR,const int& sB,double& lval,double& aval,double& bval);//============================================================================// sRGB to CIELAB conversion for 2-D images//============================================================================void DoRGBtoLABConversion(const unsigned int*& ubuff,double*& lvec,double*& avec,double*& bvec);//============================================================================// sRGB to CIELAB conversion for 3-D volumes//============================================================================void DoRGBtoLABConversion(unsigned int**& ubuff,double**& lvec,double**& avec,double**& bvec);//============================================================================// Post-processing of SLIC segmentation, to avoid stray labels.//============================================================================void EnforceLabelConnectivity(const int* labels,const int width,const int height,int*& nlabels,//input labels that need to be corrected to remove stray labelsint& numlabels,//the number of labels changes in the end if segments are removedconst int& K); //the number of superpixels desired by the user //============================================================================// Post-processing of SLIC supervoxel segmentation, to avoid stray labels.//============================================================================void EnforceSupervoxelLabelConnectivity(int**& labels,//input - previous labels, output - new labelsconst int& width,const int& height,const int& depth,const int& STEP);private:int m_width;int m_height;int m_depth;double* m_lvec;double* m_avec;double* m_bvec;double** m_lvecvec;double** m_avecvec;double** m_bvecvec;};#endif// !defined(_SLIC_H_INCLUDED_)SLICSuperpixels.h// SLICSuperpixels.h : main header file for the PROJECT_NAME application //#pragma once#ifndef __AFXWIN_H__#error "include 'stdafx.h' before including this file for PCH"#endif#include"resource.h"// main symbols// CSLICSuperpixelsApp:// See SLICSuperpixels.cpp for the implementation of this class//class CSLICSuperpixelsApp : public CWinAppEx{public:CSLICSuperpixelsApp();// Overridespublic:virtual BOOL InitInstance();// ImplementationDECLARE_MESSAGE_MAP()};extern CSLICSuperpixelsApp theApp;SLICSuperpixelsDlg.h// SLICSuperpixelsDlg.h : header file//#include<vector>#include<string>using namespace std;#pragma once// CSLICSuperpixelsDlg dialogclass CSLICSuperpixelsDlg : public CDialog{// Constructionpublic:CSLICSuperpixelsDlg(CWnd* pParent = NULL); // standard constructor// Dialog Dataenum { IDD = IDD_SLICSUPERPIXELS_DIALOG };protected:virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support // Implementationprotected:HICON m_hIcon;// Generated message map functionsvirtual BOOL OnInitDialog();afx_msg void OnPaint();afx_msg HCURSOR OnQueryDragIcon();DECLARE_MESSAGE_MAP()public:afx_msg void OnBnClickedButtonCreatesuperpixels();private:bool BrowseForFolder(string& folderpath);void GetPictures(vector<string>& picvec);void DoSupervoxelVideoSegmentation();public:afx_msg void OnEnChangeEditSpcount();afx_msg void OnEnChangeEditCompactness();int m_spcount;double m_compactness;};SLIC.cpp// SLIC.cpp: implementation of the SLIC class.//// Copyright (C) Radhakrishna Achanta 2012// All rights reserved//Email:**************************////////////////////////////////////////////////////////////////////// #include"stdafx.h"#include<cfloat>#include<cmath>#include<iostream>#include<fstream>#include"SLIC.h"//////////////////////////////////////////////////////////////////////// Construction/Destruction//////////////////////////////////////////////////////////////////////SLIC::SLIC(){m_lvec = NULL;m_avec = NULL;m_bvec = NULL;m_lvecvec = NULL;m_avecvec = NULL;m_bvecvec = NULL;}SLIC::~SLIC(){if(m_lvec) delete [] m_lvec;if(m_avec) delete [] m_avec;if(m_bvec) delete [] m_bvec;if(m_lvecvec){for( int d = 0; d < m_depth; d++ ) delete [] m_lvecvec[d];delete [] m_lvecvec;}if(m_avecvec){for( int d = 0; d < m_depth; d++ ) delete [] m_avecvec[d];delete [] m_avecvec;}if(m_bvecvec){for( int d = 0; d < m_depth; d++ ) delete [] m_bvecvec[d];delete [] m_bvecvec;}}//============================================================================== /// RGB2XYZ////// sRGB (D65 illuninant assumption) to XYZ conversion//==============================================================================void SLIC::RGB2XYZ(const int& sR,const int& sG,const int& sB,double& X,double& Y,double& Z){double R = sR/255.0;double G = sG/255.0;double B = sB/255.0;double r, g, b;if(R <= 0.04045) r = R/12.92;else r = pow((R+0.055)/1.055,2.4);if(G <= 0.04045) g = G/12.92;else g = pow((G+0.055)/1.055,2.4);if(B <= 0.04045) b = B/12.92;else b = pow((B+0.055)/1.055,2.4);X = r*0.4124564 + g*0.3575761 + b*0.1804375;Y = r*0.2126729 + g*0.7151522 + b*0.0721750;Z = r*0.0193339 + g*0.1191920 + b*0.9503041;}//===========================================================================/// RGB2LAB//===========================================================================void SLIC::RGB2LAB(const int& sR, const int& sG, const int& sB, double& lval, double& aval, double& bval){//------------------------// sRGB to XYZ conversion//------------------------double X, Y, Z;RGB2XYZ(sR, sG, sB, X, Y, Z);//------------------------// XYZ to LAB conversion//------------------------double epsilon = 0.008856; //actual CIE standarddouble kappa = 903.3; //actual CIE standarddouble Xr = 0.950456; //reference whitedouble Yr = 1.0; //reference whitedouble Zr = 1.088754; //reference whitedouble xr = X/Xr;double yr = Y/Yr;double zr = Z/Zr;double fx, fy, fz;if(xr > epsilon) fx = pow(xr, 1.0/3.0);else fx = (kappa*xr + 16.0)/116.0;if(yr > epsilon) fy = pow(yr, 1.0/3.0);else fy = (kappa*yr + 16.0)/116.0;if(zr > epsilon) fz = pow(zr, 1.0/3.0);else fz = (kappa*zr + 16.0)/116.0;lval = 116.0*fy-16.0;aval = 500.0*(fx-fy);bval = 200.0*(fy-fz);}//=========================================================================== /// DoRGBtoLABConversion////// For whole image: overlaoded floating point version//=========================================================================== void SLIC::DoRGBtoLABConversion(const unsigned int*& ubuff,double*& lvec,double*& avec,double*& bvec){int sz = m_width*m_height;lvec = new double[sz];avec = new double[sz];bvec = new double[sz];for( int j = 0; j < sz; j++ ){int r = (ubuff[j] >> 16) & 0xFF;int g = (ubuff[j] >> 8) & 0xFF;int b = (ubuff[j] ) & 0xFF;RGB2LAB( r, g, b, lvec[j], avec[j], bvec[j] );}}//===========================================================================/// DoRGBtoLABConversion////// For whole volume//===========================================================================void SLIC::DoRGBtoLABConversion(unsigned int**& ubuff,double**& lvec,double**& avec,double**& bvec){int sz = m_width*m_height;for( int d = 0; d < m_depth; d++ ){for( int j = 0; j < sz; j++ ){int r = (ubuff[d][j] >> 16) & 0xFF;int g = (ubuff[d][j] >> 8) & 0xFF;int b = (ubuff[d][j] ) & 0xFF;RGB2LAB( r, g, b, lvec[d][j], avec[d][j], bvec[d][j] );}}}//================================================================================= /// DrawContoursAroundSegments////// Internal contour drawing option exists. One only needs to comment the if/// statement inside the loop that looks at neighbourhood.//=================================================================================void SLIC::DrawContoursAroundSegments(unsigned int*& ubuff,int*& labels,const int& width,const int& height,const unsigned int& color ){const int dx8[8] = {-1, -1, 0, 1, 1, 1, 0, -1};const int dy8[8] = { 0, -1, -1, -1, 0, 1, 1, 1};/* int sz = width*height;vector<bool> istaken(sz, false);int mainindex(0);for( int j = 0; j < height; j++ ){for( int k = 0; k < width; k++ ){int np(0);for( int i = 0; i < 8; i++ ){int x = k + dx8[i];int y = j + dy8[i];if( (x >= 0 && x < width) && (y >= 0 && y < height) ){int index = y*width + x;if( false == istaken[index] )//comment this to obtain internal contours{if( labels[mainindex] != labels[index] ) np++;}}}if( np > 1 )//change to 2 or 3 for thinner lines{ubuff[mainindex] = color;istaken[mainindex] = true;}mainindex++;}}*/int sz = width*height;vector<bool> istaken(sz, false);vector<int> contourx(sz);vector<int> contoury(sz);int mainindex(0);int cind(0);for( int j = 0; j < height; j++ ){for( int k = 0; k < width; k++ ){int np(0);for( int i = 0; i < 8; i++ ){int x = k + dx8[i];int y = j + dy8[i];if( (x >= 0 && x < width) && (y >= 0 && y < height) ){int index = y*width + x;//if( false == istaken[index] )//comment this to obtain internal contours{if( labels[mainindex] != labels[index] ) np++;}}}if( np > 1 ){contourx[cind] = k;contoury[cind] = j;istaken[mainindex] = true;//img[mainindex] = color;cind++;}mainindex++;}}int numboundpix = cind;//int(contourx.size());for( int j = 0; j < numboundpix; j++ ){int ii = contoury[j]*width + contourx[j];ubuff[ii] = 0xffffff;for( int n = 0; n < 8; n++ ){int x = contourx[j] + dx8[n];int y = contoury[j] + dy8[n];if( (x >= 0 && x < width) && (y >= 0 && y < height) ){int ind = y*width + x;if(!istaken[ind]) ubuff[ind] = 0;}}}}//============================================================================== /// DetectLabEdges//============================================================================== void SLIC::DetectLabEdges(const double* lvec,const double* avec,const double* bvec,const int& width,const int& height,vector<double>& edges){int sz = width*height;edges.resize(sz,0);for( int j = 1; j < height-1; j++ ){for( int k = 1; k < width-1; k++ ){int i = j*width+k;double dx = (lvec[i-1]-lvec[i+1])*(lvec[i-1]-lvec[i+1]) +(avec[i-1]-avec[i+1])*(avec[i-1]-avec[i+1]) +(bvec[i-1]-bvec[i+1])*(bvec[i-1]-bvec[i+1]);double dy = (lvec[i-width]-lvec[i+width])*(lvec[i-width]-lvec[i+width]) +(avec[i-width]-avec[i+width])*(avec[i-width]-avec[i+width]) +(bvec[i-width]-bvec[i+width])*(bvec[i-width]-bvec[i+width]);//edges[i] = fabs(dx) + fabs(dy);edges[i] = dx*dx + dy*dy;}}}//===========================================================================/// PerturbSeeds//===========================================================================void SLIC::PerturbSeeds(vector<double>& kseedsl,vector<double>& kseedsa,vector<double>& kseedsb,vector<double>& kseedsx,vector<double>& kseedsy,const vector<double>& edges){const int dx8[8] = {-1, -1, 0, 1, 1, 1, 0, -1};const int dy8[8] = { 0, -1, -1, -1, 0, 1, 1, 1};int numseeds = kseedsl.size();for( int n = 0; n < numseeds; n++ ){int ox = kseedsx[n];//original xint oy = kseedsy[n];//original yint oind = oy*m_width + ox;int storeind = oind;for( int i = 0; i < 8; i++ ){int nx = ox+dx8[i];//new xint ny = oy+dy8[i];//new yif( nx >= 0 && nx < m_width && ny >= 0 && ny < m_height){int nind = ny*m_width + nx;if( edges[nind] < edges[storeind]){storeind = nind;}}}if(storeind != oind){kseedsx[n] = storeind%m_width;kseedsy[n] = storeind/m_width;kseedsl[n] = m_lvec[storeind];kseedsa[n] = m_avec[storeind];kseedsb[n] = m_bvec[storeind];}}}//=========================================================================== /// GetLABXYSeeds_ForGivenStepSize////// The k seed values are taken as uniform spatial pixel samples.//=========================================================================== void SLIC::GetLABXYSeeds_ForGivenStepSize(vector<double>& kseedsl,vector<double>& kseedsa,vector<double>& kseedsb,vector<double>& kseedsx,vector<double>& kseedsy,const int& STEP,const bool& perturbseeds,const vector<double>& edgemag){const bool hexgrid = false;int numseeds(0);int n(0);//int xstrips = m_width/STEP;//int ystrips = m_height/STEP;int xstrips = (0.5+double(m_width)/double(STEP));int ystrips = (0.5+double(m_height)/double(STEP));int xerr = m_width - STEP*xstrips;if(xerr < 0){xstrips--;xerr = m_width - STEP*xstrips;} int yerr = m_height - STEP*ystrips;if(yerr < 0){ystrips--;yerr = m_height- STEP*ystrips;}double xerrperstrip = double(xerr)/double(xstrips);double yerrperstrip = double(yerr)/double(ystrips);int xoff = STEP/2;int yoff = STEP/2;//-------------------------numseeds = xstrips*ystrips;//-------------------------kseedsl.resize(numseeds);kseedsa.resize(numseeds);kseedsb.resize(numseeds);kseedsx.resize(numseeds);kseedsy.resize(numseeds);for( int y = 0; y < ystrips; y++ ){int ye = y*yerrperstrip;for( int x = 0; x < xstrips; x++ ){int xe = x*xerrperstrip;int seedx = (x*STEP+xoff+xe);if(hexgrid){ seedx = x*STEP+(xoff<<(y&0x1))+xe; seedx =min(m_width-1,seedx); }//for hex grid samplingint seedy = (y*STEP+yoff+ye);int i = seedy*m_width + seedx;kseedsl[n] = m_lvec[i];kseedsa[n] = m_avec[i];kseedsb[n] = m_bvec[i];kseedsx[n] = seedx;kseedsy[n] = seedy;n++;}}if(perturbseeds){PerturbSeeds(kseedsl, kseedsa, kseedsb, kseedsx, kseedsy, edgemag);}}//===========================================================================/// GetKValues_LABXYZ////// The k seed values are taken as uniform spatial pixel samples.//===========================================================================void SLIC::GetKValues_LABXYZ(vector<double>& kseedsl,vector<double>& kseedsa,vector<double>& kseedsb,vector<double>& kseedsx,vector<double>& kseedsy,vector<double>& kseedsz,const int& STEP){const bool hexgrid = false;int numseeds(0);int n(0);int xstrips = (0.5+double(m_width)/double(STEP));int ystrips = (0.5+double(m_height)/double(STEP));int zstrips = (0.5+double(m_depth)/double(STEP));int xerr = m_width - STEP*xstrips;if(xerr < 0){xstrips--;xerr = m_width - STEP*xstrips;} int yerr = m_height - STEP*ystrips;if(yerr < 0){ystrips--;yerr = m_height- STEP*ystrips;} int zerr = m_depth - STEP*zstrips;if(zerr < 0){zstrips--;zerr = m_depth - STEP*zstrips;}double xerrperstrip = double(xerr)/double(xstrips);double yerrperstrip = double(yerr)/double(ystrips);double zerrperstrip = double(zerr)/double(zstrips);int xoff = STEP/2;int yoff = STEP/2;int zoff = STEP/2;//-------------------------numseeds = xstrips*ystrips*zstrips;//-------------------------kseedsl.resize(numseeds);kseedsa.resize(numseeds);kseedsb.resize(numseeds);kseedsx.resize(numseeds);kseedsy.resize(numseeds);kseedsz.resize(numseeds);for( int z = 0; z < zstrips; z++ ){int ze = z*zerrperstrip;int d = (z*STEP+zoff+ze);for( int y = 0; y < ystrips; y++ ){int ye = y*yerrperstrip;for( int x = 0; x < xstrips; x++ ){int xe = x*xerrperstrip;int i = (y*STEP+yoff+ye)*m_width + (x*STEP+xoff+xe);kseedsl[n] = m_lvecvec[d][i];kseedsa[n] = m_avecvec[d][i];kseedsb[n] = m_bvecvec[d][i];kseedsx[n] = (x*STEP+xoff+xe);kseedsy[n] = (y*STEP+yoff+ye);kseedsz[n] = d;n++;}}}}//=========================================================================== /// PerformSuperpixelSLIC////// Performs k mean segmentation. It is fast because it looks locally, not/// over the entire image.//=========================================================================== void SLIC::PerformSuperpixelSLIC(vector<double>& kseedsl,vector<double>& kseedsa,vector<double>& kseedsb,vector<double>& kseedsx,vector<double>& kseedsy,int*& klabels,const int& STEP,const vector<double>& edgemag,const double& M){int sz = m_width*m_height;const int numk = kseedsl.size();//----------------int offset = STEP;//if(STEP < 8) offset = STEP*1.5;//to prevent a crash due to a very small step size //----------------vector<double> clustersize(numk, 0);vector<double> inv(numk, 0);//to store 1/clustersize[k] valuesvector<double> sigmal(numk, 0);vector<double> sigmaa(numk, 0);vector<double> sigmab(numk, 0);vector<double> sigmax(numk, 0);vector<double> sigmay(numk, 0);vector<double> distvec(sz, DBL_MAX);double invwt = 1.0/((STEP/M)*(STEP/M));int x1, y1, x2, y2;double l, a, b;double dist;double distxy;for( int itr = 0; itr < 10; itr++ ){distvec.assign(sz, DBL_MAX);for( int n = 0; n < numk; n++ ){y1 = max(0.0, kseedsy[n]-offset);y2 = min((double)m_height, kseedsy[n]+offset);x1 = max(0.0, kseedsx[n]-offset);。