drwnMultiSegConfig.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:    drwnMultiSegConfig.h
** AUTHOR(S):   Stephen Gould <stephen.gould@anu.edu.au>
**
*****************************************************************************/

#pragma once

#include <cstdlib>
#include <cassert>
#include <list>
#include <map>

#include "Eigen/Core"

#include "cv.h"

#include "drwnBase.h"

using namespace std;

// drwnMultiSegConfig -------------------------------------------------------

class drwnMultiSegConfig : public drwnProperties, drwnConfigurableModule {
 protected:
    string _baseDir;        
    string _imgDir;         
    string _lblDir;         
    string _segDir;         
    string _cacheDir;       
    string _modelsDir;      
    string _outputDir;      
    string _imgExt;         
    string _lblExt;         
    string _segExt;         
    bool _bUseCache;        
    bool _bCompressedCache; 

 public:
    drwnMultiSegConfig();
    ~drwnMultiSegConfig();

    // options i/o
    void readConfiguration(drwnXMLNode& node);
    void setConfiguration(const char *name, const char *value) {
        drwnProperties::setProperty(drwnProperties::findProperty(name), value);
    }

    void usage(std::ostream& os) const;

    // utility functions
    string filebase(const char *dirKey, const char *baseName);
    string filebase(const char *dirKey, const string& baseName);
    string filename(const char *dirKey, const char *baseName, const char *extKey);
    string filename(const char *dirKey, const string& baseName, const char *extKey);
};

extern drwnMultiSegConfig gMultiSegConfig;

// drwnStandardMultiSegRegionDatasets ----------------------------------------

typedef enum _drwnStandardMultiSegRegionDatasets {
    DRWN_DS_FGBG,     
    DRWN_DS_MSRC,     
    DRWN_DS_STANFORD, 
    DRWN_DS_PASCALVOC 
} drwnStandardMultiSegRegionDatasets;

// drwnMultiSegRegionDefinitions ---------------------------------------------

class drwnMultiSegRegionDefinitions {
 protected:
    map<int, int> _keys;          
    vector<unsigned int> _colors; 
    vector<string> _names;        

 public:
    drwnMultiSegRegionDefinitions();
    drwnMultiSegRegionDefinitions(const char *filename);
    virtual ~drwnMultiSegRegionDefinitions();

    // i/o
    void clear();
    void read(const char *filename);
    void read(drwnXMLNode& root);

    bool empty() const { return _colors.empty(); }
    int size() const { return (int)_colors.size(); }
    int maxKey() const;

    set<int> keys() const;
    string name(int key) const;
    unsigned int color(int key) const;

    static unsigned char red(unsigned int c) {
        return ((c >> 16) & 0xff);
    }
    static unsigned char green(unsigned int c) {
        return ((c >> 8) & 0xff);
    }
    static unsigned char blue(unsigned int c) { 
        return (c & 0xff);
    }
    static unsigned int rgb(unsigned char r, unsigned char g, unsigned char b) {
        return ((unsigned int)(r << 16) | (unsigned int)(g << 8) | (unsigned int)b);
    }

    virtual void initializeForDataset(const drwnStandardMultiSegRegionDatasets& dataset);

    MatrixXi convertImageToLabels(const cv::Mat& img) const;
    MatrixXi convertImageToLabels(const char *filename) const;

    MatrixXi loadLabelFile(const char *filename) const;

    // operators
    unsigned int& operator[](int key);

 protected:
    virtual void initDefaultRegions();
};

extern drwnMultiSegRegionDefinitions gMultiSegRegionDefs;