vc_opencv.h

Go to the documentation of this file.

#ifndef VC_OPENCV__H
#define VC_OPENCV__H

#include "VCells.h"
#include <opencv2/opencv.hpp>

class VC_OpenCV {
public:
    static void computeSuperpixels(const cv::Mat &image, int superpixels, 
            double weight_length, int radius, int num_nei_cluster, 
            int num_direct_nei, int threshold, cv::Mat &labels) {
        
        // i: height, j: width, k: channels
        // (pBmpBuf + i * lineByte + j * 3 + k)
        VCells vc(superpixels, weight_length);
//  vc.pBmpBuf = new unsigned char[image.rows*image.cols*image.channels()];
        vc.bmpWidth = image.cols;
        vc.bmpHeight = image.rows;
        vc.lineByte = vc.bmpWidth;
        // lineByte = (bmpWidth * biBitCount / 8 + 3) / 4 * 4;
        
    struct pixel* pixelArray = new pixel[vc.bmpHeight*vc.bmpWidth];
        for (int i = 0; i < vc.bmpHeight; i++) {
            for (int j = 0; j < vc.bmpWidth; j++) {
                int index = vc.getIndexFromRC(i, j);
                pixelArray[index].color[0] = image.at<cv::Vec3b>(i, j)[0];
                pixelArray[index].color[1] = image.at<cv::Vec3b>(i, j)[1];
                pixelArray[index].color[2] = image.at<cv::Vec3b>(i, j)[2];
            }
    }
        
    /************************************************************************************/
    /************                     initialization                         ************/
    /************************************************************************************/
    struct centroid* generators = new centroid[superpixels];
    vc.initializePixel(pixelArray);
    vc.initializeGenerators(generators, pixelArray); 
    vc.classicCVT(pixelArray, generators);

    /************************************************************************************/
    /************                     VCells by EWCVT                        ************/
    /************************************************************************************/
    vc.EWCVT(generators, pixelArray);
        
        labels.create(image.rows, image.cols, CV_32SC1);
        for (int i = 0; i < vc.bmpHeight; i++) {
            for (int j = 0; j < vc.bmpWidth; j++) {
                int index = vc.getIndexFromRC(i, j);
                labels.at<int>(i, j) = pixelArray[index].indexCluster;
            }
    }
        
    delete[] pixelArray;
    delete[] generators;
    }
};

#endif  /* VC_OPENCV__H */