ergc_opencv.h

Go to the documentation of this file.

#ifndef ERGC_OPENCV_H
#define ERGC_OPENCV_H

#include <CImg.h>
#include <opencv2/opencv.hpp>
#include "ergc.h"

class ERGC_OpenCV {
public:
    static void computeSuperpixels(const cv::Mat &image, int region_height, int region_width, 
            bool lab, bool perturb_seeds, int m, cv::Mat &labels) {
        
        int dx = region_width; // Seeds sampling wrt axis x (for custom grids)
        int dy = region_height; // Seeds sampling wrt axis y (for custom grids)
        // int m = 0; // Compacity value

        CImg<> im(image.cols, image.rows, 1, 3);
        for (int i = 0; i < image.rows; i++) {
            for (int j = 0; j < image.cols; j++) {
                im(j, i, 0, 0) = image.at<cv::Vec3b>(i, j)[0];
                im(j, i, 0, 1) = image.at<cv::Vec3b>(i, j)[1];
                im(j, i, 0, 2) = image.at<cv::Vec3b>(i, j)[2];
            }
        }
        
        // useful variables
        CImg<> distances;
        CImg<int> states, seeds;
        vector<SP*> SPs;

        // convert to Lab if needed (better superpixels with color images)
        if (lab) {
            im.RGBtoLab();
        }

        CImg<> gradient;
        if (perturb_seeds) {
            gradient = compute_gradient(im);
        }

        placeSeedsOnCustomGrid2d(im.width(), im.height(), dx, dy, seeds);

        if (perturb_seeds) {
            CImg<int> perturbedSeeds;
            if (im.depth() == 1) {
                perturbSeeds2d(seeds, gradient, perturbedSeeds);
            }
            else {
                perturbSeeds3d(seeds, gradient, perturbedSeeds);
            }
            
            seeds = perturbedSeeds;
        }

        initialize_images(seeds, distances, states);
        SPs = initialize_superpixels(im, seeds);

        fmm3d(distances, seeds, states, im, SPs, m);

        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) = seeds(j, i, 0, 0);
            }
        }
    }
};

#endif  /* ERGC_OPENCV_H */