crs_opencv.h

Go to the documentation of this file.

#ifndef CRS_OPENCV_H
#define CRS_OPENCV_H

#include <opencv2/opencv.hpp>
#include "FeatureType.h"
#include "ContourRelaxation.h"
#include "InitializationFunctions.h"

class CRS_OpenCV {
public:
    static void computeSuperpixels(const cv::Mat &image, int region_height, 
            int region_width, double clique_cost, double compactness, 
            int iterations, int color_space, cv::Mat &labels) {
        
        double diagonal_cost = clique_cost/std::sqrt(2);
        
        bool color_image = false;
        if (image.channels() == 3) {
            color_image = true;
        }
        
        std::vector<FeatureType> features;
        if (color_image) {
            features.push_back(Color);
        }
        else {
            features.push_back(Grayvalue);
        }

        features.push_back(Compactness);
        
        ContourRelaxation<boost::uint16_t> contour_relaxation(features);
        contour_relaxation.setCompactnessData(compactness);
        
        if (color_image) {
            
            cv::Mat image_YCrCb;
            std::vector<cv::Mat> image_channels;
            
            switch (color_space) {
                default:
                case 0: // YCrCb
                    cv::cvtColor(image, image_YCrCb, CV_BGR2YCrCb);
                    cv::split(image_YCrCb, image_channels);
                    break;
                case 1: // RGB
                    cv::split(image, image_channels);
                    break;
            }

            contour_relaxation.setColorData(image_channels[0], image_channels[1], 
                    image_channels[2]);
        }
        else {
//            cv::Mat imageGray = image.clone();
//            cv::cvtColor(imageGray, image, CV_GRAY2BGR);

            contour_relaxation.setGrayvalueData(image);
        }

        cv::Mat label_image = createBlockInitialization<boost::uint16_t>(image.size(), 
                region_width, region_height);
        cv::Mat relaxed_label_image;
        cv::Mat mean_image;
        
        contour_relaxation.relax(label_image, clique_cost, diagonal_cost, 
                iterations, relaxed_label_image, mean_image);
        
        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) = relaxed_label_image.at<boost::uint16_t>(i, j);
            }
        }
    }
};

#endif  /* CRS_OPENCV_H */