NICE_Segmentation/RSGraphBased.h

/**
 * @file RSGraphBased.h
 * @brief fast Segmentation Method based on Felzenszwalb2004
 * @author Björn Fröhlich
 * @date 02/22/2010
 */
#ifndef RSGraphBasedINCLUDE
#define RSGraphBasedINCLUDE

#include "core/basics/Config.h"
#include "RegionSegmentationMethod.h"

namespace OBJREC {

class RSGraphBased: public RegionSegmentationMethod
{

  protected:
    double parameter_k;
    int parameter_min_size;
    double parameter_sigma;

  public:
    /** simple constructor */
    RSGraphBased();

    /** simple constructor */
    RSGraphBased(double t);

    /**
     * standard constructor
     * @param conf config file
     */
    RSGraphBased(const NICE::Config *conf );

    /** simple destructor */
    ~RSGraphBased();

    /**
     * returns the regions of a given image
     * @param img input image
     * @param mask output regions, each region has it own number
     * @return count of region
     */
    virtual int segRegions ( const NICE::Image & img, NICE::Matrix & mask ) const;

    /**
     * returns the regions of a given image
     * @param img input color image
     * @param mask output regions, each region has it own number
     * @return count of region
     */
    virtual int segRegions ( const NICE::ColorImage & img, NICE::Matrix & mask) const;
};

} //namespace

#endif