fh_opencv.h

Go to the documentation of this file.

#ifndef FH_OPENCV_H
#define FH_OPENCV_H

#include <opencv2/opencv.hpp>
#include "misc.h"
#include "image.h"
#include "segment-image-labels.h"

class FH_OpenCV {
public:
    static int computeSuperpixels(const cv::Mat &mat, float sigma, 
            float threshold, int minimum_size, cv::Mat &labels) {
        
        image<rgb>* rgbImage = new image<rgb>(mat.cols, mat.rows);
        
        for (int i = 0; i < mat.rows; ++i) {
            for (int j = 0; j < mat.cols; ++j) {
                imRef(rgbImage, j, i).r = mat.at<cv::Vec3b>(i, j)[2];
                imRef(rgbImage, j, i).g = mat.at<cv::Vec3b>(i, j)[1];
                imRef(rgbImage, j, i).b = mat.at<cv::Vec3b>(i, j)[0];
            }
        }
        
        int superpixels = 0;
        image<int> *segmentation = segment_image_labels(rgbImage, sigma, threshold, minimum_size, &superpixels);

        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) = imRef(segmentation, j, i);
            }
        }
        
        delete rgbImage;
        delete segmentation;
        
        return superpixels;
    }
};

#endif  /* FH_OPENCV_H */