SuperpixelModuleSEEDS.h

#ifndef SUPERPIXELMODULESEEDS_H
#define SUPERPIXELMODULESEEDS_H

#include <map>
#include <string>
#include <QImage>
#include "Datapool.h"
#include "ModuleInterface.h"
#include <QDir>
#include "src/json.h"
#include "ReliabilityBg.h"
#include "opencv2/opencv.hpp"
#ifdef __OPENCV3__
#include "src/ximgproc/ximgproc.hpp"
#endif

class SuperpixelModuleSEEDS: public ModuleInterface {
public:
    SuperpixelModuleSEEDS(Datapool *i_data);
    ~SuperpixelModuleSEEDS();

    //Set module configuration parameters
    bool setParameters(QDomNode& config);

    //Initialization after reading parameters
    bool init();

    //Function executed at each frame
    bool run();

    //update parameters at runtime.
    bool updateParameters();

    //Suppress extra pixel space per line
    uchar *suppressExtraPixelSpace(QImage *image);
#ifdef __OPENCV3__
    cv::Ptr<cv::ximgproc::SuperpixelSEEDS> m_SEEDS;
#endif
    //Desired number of superpixels. Note that the actual number may be smaller due to restrictions (depending on the image size and num_levels).
    //Use getNumberOfSuperpixels() to get the actual number.
    int m_superpixelnumber;
    //Number of block levels. The more levels, the more accurate is the segmentation, but needs more memory and CPU time.
    int m_num_levels;
    //Use prior – enable 3x3 shape smoothing term if >0. A larger value leads to smoother shapes. prior must be in the range [0, 5].
    int m_use_prior;
    //Number of histogram bins.
    int m_histogram_bins;
    //If true, iterate each block level twice for higher accuracy.
    bool m_double_step;
    //Number of pixel level iterations. Higher number improves the result.
    int m_num_iterations;


    private:
        int currentFrame;
        int line;
        bool first;
        bool validate_once;
        bool different_bg_current;
};

#endif // SUPERPIXELMODULESEEDS_H