lsc_opencv.h

Go to the documentation of this file.

#ifndef LSC_OPENCV_H
#define LSC_OPENCV_H

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

class LSC_OpenCV {
public:
    static void computeSuperpixels(const cv::Mat &image, int region_height, 
            int region_width, double ratio, int iterations, int threshold, 
            int color_space, cv::Mat &labels)
    {
        unsigned char* R = new unsigned char[image.rows*image.cols];
        unsigned char* G = new unsigned char[image.rows*image.cols];
        unsigned char* B = new unsigned char[image.rows*image.cols];
        
        for (int i = 0; i < image.rows; i++) {
            for (int j = 0; j < image.cols; j++) {
                R[i*image.cols + j] = image.at<cv::Vec3b>(i, j)[2];
                G[i*image.cols + j] = image.at<cv::Vec3b>(i, j)[1];
                B[i*image.cols + j] = image.at<cv::Vec3b>(i, j)[0];
            }
        }
        
        unsigned short* labeling = new unsigned short[image.rows*image.cols];
        for (int i = 0; i < image.rows*image.cols; i++) {
            labeling[i] = 0;
        }
        
        LSC(R, G, B, image.rows, image.cols, region_height, region_width, ratio, 
                iterations, threshold, color_space, labeling);
        
        labels.create(image.rows, image.cols, CV_32SC1);
        for (int i = 0; i < image.rows; i++) {
            for (int j = 0; j < image.cols; j++) {
                labels.at<int>(i, j) = labeling[i*image.cols + j];
            }
        }
    }
};

#endif  /* LSC_OPENCV_H */