vlslic_opencv.h

Go to the documentation of this file.

#ifndef VLSLIC_OPENCV_H
#define VLSLIC_OPENCV_H

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

class VLSLIC_OpenCV {
public:
    static void computeSuperpixels(const cv::Mat &mat, int region_size, 
            double regularization, int min_region_size, int iterations, cv::Mat &labels)
    {
        // Convert image to one-dimensional array.
        float* image = new float[mat.rows*mat.cols*mat.channels()];
        for (int i = 0; i < mat.rows; ++i) {
            for (int j = 0; j < mat.cols; ++j) {
                if (mat.channels() == 1) {
                    image[j + mat.cols*i] = mat.at<unsigned char>(i, j);
                }
                else if (mat.channels() == 3) {
                    image[j + mat.cols*i + mat.cols*mat.rows*0] = mat.at<cv::Vec3b>(i, j)[0];
                    image[j + mat.cols*i + mat.cols*mat.rows*1] = mat.at<cv::Vec3b>(i, j)[1];
                    image[j + mat.cols*i + mat.cols*mat.rows*2] = mat.at<cv::Vec3b>(i, j)[2];
                }
            }
        }
        
        vl_uint32* segmentation = new vl_uint32[mat.rows*mat.cols];
        vl_size height = mat.rows;
        vl_size width = mat.cols;
        vl_size channels = mat.channels();
        
        vl_slic_segment_t(segmentation, image, width, height, channels, region_size, 
                regularization, min_region_size, iterations);
        
        // Convert segmentation.
        labels.create(mat.rows, mat.cols, CV_32SC1);
        for (int i = 0; i < mat.rows; ++i) {
            for (int j = 0; j < mat.cols; ++j) {
                labels.at<int>(i, j) = (int) segmentation[j + mat.cols*i];
            }
        }
    }
};

#endif  /* VLSLIC_OPENCV_H */