drwnSegImageInstance.h

/*****************************************************************************
** DARWIN: A FRAMEWORK FOR MACHINE LEARNING RESEARCH AND DEVELOPMENT
** Distributed under the terms of the BSD license (see the LICENSE file)
** Copyright (c) 2007-2015, Stephen Gould
** All rights reserved.
**
******************************************************************************
** FILENAME:    drwnSegImageInstance.h
** AUTHOR(S):   Stephen Gould <stephen.gould@anu.edu.au>
**
*****************************************************************************/

#pragma once

#include <cstdlib>
#include <cassert>
#include <iostream>
#include <fstream>
#include <iomanip>
#include <vector>
#include <list>

#include "Eigen/Core"

#include "drwnBase.h"
#include "drwnIO.h"
#include "drwnML.h"

#include "drwnPixelNeighbourContrasts.h"
#include "drwnSuperpixelContainer.h"
#include "drwnSegImagePixelFeatures.h"
#include "drwnVisionUtils.h"

using namespace std;
using namespace Eigen;

// drwnSegImageInstance class ----------------------------------------------

class drwnSegImageInstance {
 protected:
    string _baseName;                   
    cv::Mat _img;                       
    cv::Mat _grayImg;                   
    cv::Mat _softEdgeImg;               

 public:
    // cached pixel features
    vector<vector<double> > unaries;         
    drwnPixelNeighbourContrasts contrast;    

    // long range edges
    vector<drwnWeightedPixelEdge> auxEdges;  

    // superpixels and auxiliary data
    drwnSuperpixelContainer superpixels;     
    vector<cv::Mat> auxiliaryData;           

    // class labels
    MatrixXi pixelLabels;                    

 public:
    drwnSegImageInstance(const char *imgFilename, const char *baseName = NULL);
    drwnSegImageInstance(const cv::Mat& img, const char *baseName = NULL);
    drwnSegImageInstance(const drwnSegImageInstance& instance);
    virtual ~drwnSegImageInstance();

    inline const cv::Mat& image() const { return _img; }
    inline const cv::Mat& greyImage() const { return _grayImg; }
    inline const cv::Mat& edgeMap() const { return _softEdgeImg; }
    inline const string &name() const { return _baseName; }

    inline int width() const { return _img.cols; }
    inline int height() const { return _img.rows; }
    inline int size() const { return (_img.cols * _img.rows); }
    //void resize(int w, int h);

    inline int pixel2Indx(const cv::Point& p) const { return pixel2Indx(p.x, p.y); }
    inline int pixel2Indx(int x, int y) const { return y * _img.cols + x; }
    inline cv::Point indx2Pixel(int indx) const {
        return cv::Point(indx % _img.cols, indx / _img.cols);
    }

    void clearPixelFeatures();
    inline void appendPixelFeatures() {
        drwnSegImageStdPixelFeatures g;
        appendPixelFeatures(g);
    }
    void appendPixelFeatures(drwnSegImagePixelFeatures &featureGenerator);

    drwnSegImageInstance& operator=(const drwnSegImageInstance& instance);

 protected:
    void initInstance();
};