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

1、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 defau

2、lt 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 f

3、iles,/ 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_CSTRIN

4、G_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_

5、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 barsSL

6、IC.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

7、 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

8、 size = step*step)/= void DoSuperpixelSegmentation_ForGivenSuperpixelSize( const unsigned int* ubuff,/Each 32 bit unsigned int contains ARGB pixel values.const intwidth,const intheight,int*&klabels,int&numlabels, const int&superpixelsize, const double& compactness);/=/ Superpixel seg

9、mentation for a given number of superpixels/= void DoSuperpixelSegmentation_ForGivenNumberOfSuperpixels( const unsigned int* ubuff,const intwidth,const intheight,int*&klabels,int&numlabels, const int&K,/required number of superpixels const double& compactness);/10-20 is a good value

10、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&a

11、mp; 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 SaveS

12、upervoxelLabels(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 lay

13、er./=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&g

14、t;&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(

15、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

16、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<

17、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);/=/

18、 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<d

19、ouble>&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,cons

20、t 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&sG,const int&sB,double&lval,double&aval,double&bval);/=/ sRGB to CIELAB conversion for 2-D images/=void DoRGBtoLABC

21、onversion(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 la

22、bels./=void EnforceLabelConnectivity(const int*labels,const intwidth,const intheight,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 b

23、y 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,int&numlabels,const int&STEP);private:intm_

24、width;intm_height;intm_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 &#

25、39;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 CWinAppExpublic:CSLICSuperpixelsApp();/ Overridespublic:virtual BOOL Ini

26、tInstance();/ 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/ Construct

27、ionpublic: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(

28、);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 On

29、EnChangeEditSpcount();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: firstname.lastnameepfl.ch/#include "stdafx.h"#include <cfloat>

30、#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)

31、 delete m_bvec;if(m_lvecvec)for( int d = 0; d < m_depth; d+ ) delete m_lvecvecd;delete m_lvecvec;if(m_avecvec)for( int d = 0; d < m_depth; d+ ) delete m_avecvecd;delete m_avecvec;if(m_bvecvec)for( int d = 0; d < m_depth; d+ ) delete m_bvecvecd;delete m_bvecvec;/=/RGB2XYZ/ sRGB (D65 illunina

32、nt 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;elser = pow(R+0.055)/1.055,2.4);if(G <= 0.04045)g

33、 = G/12.92;elseg = pow(G+0.055)/1.055,2.4);if(B <= 0.04045)b = B/12.92;elseb = 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&

34、; 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 whit

35、edouble 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);elsefx = (kappa*xr + 16.0)/116.0;if(yr > epsilon)fy = pow(yr, 1.0/3.0);elsefy = (kappa*yr + 16.0)/116.0;if(zr > ep

36、silon)fz = pow(zr, 1.0/3.0);elsefz = (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,doubl

37、e*&bvec)int sz = m_width*m_height;lvec = new doublesz;avec = new doublesz;bvec = new doublesz;for( int j = 0; j < sz; j+ )int r = (ubuffj >> 16) & 0xFF;int g = (ubuffj >> 8) & 0xFF;int b = (ubuffj ) & 0xFF;RGB2LAB( r, g, b, lvecj, avecj, bvecj );/=/DoRGBtoLABConversion

38、/ 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 = (ubuffdj >> 16) & 0xFF;int g = (ubuffdj >> 8) & 0xFF;

39、int b = (ubuffdj ) & 0xFF;RGB2LAB( r, g, b, lvecdj, avecdj, bvecdj );/=/ 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*

40、&labels,const int&width,const int&height,const unsigned int&color )const int dx88 = -1, -1, 0, 1, 1, 1, 0, -1;const int dy88 = 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 + dx