drwnVisionUtils.h

Go to the documentation of this file.

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

#pragma once

#include <cstdlib>
#include <cassert>
#include <vector>

#include "Eigen/Core"

#include "cv.h"

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

using namespace std;
using namespace Eigen;

class drwnWeightedPixelEdge {
 public:
    double w;     
    cv::Point p;  
    cv::Point q;  

 public:
    drwnWeightedPixelEdge() : w(0.0), p(-1, -1), q(-1, -1) { /* do nothing */ };
    drwnWeightedPixelEdge(const cv::Point& _p, const cv::Point& _q, double _w = 0.0) :
        w(_w), p(_p), q(_q) { /* do nothing */ };
    ~drwnWeightedPixelEdge() { /* do nothing */ };

    bool operator<(const drwnWeightedPixelEdge& e) const {
        if (w < e.w) return true;
        if (w > e.w) return false;
        if (p.y < e.p.y) return true;
        if (p.y > e.p.y) return false;
        if (p.x < e.p.x) return true;
        if (p.x > e.p.x) return false;
        if (q.y < e.q.y) return true;
        if (q.y > e.q.y) return false;
        return (q.x < e.q.x);
    };
};

inline double drwnPyramidScale(int lambda) {
    return exp(-1.0 * log(2.0) / (double)lambda);
}

cv::Rect drwnTransformROI(const cv::Rect& roi, const cv::Size& srcSize, const cv::Size& dstSize);

void drwnLoadPixelLabels(cv::Mat& pixelLabels, const char *filename);

void drwnLoadPixelLabels(cv::Mat& pixelLabels, const char *filename, int numLabels);

void drwnLoadPixelLabels(MatrixXi &pixelLabels, const char *filename,
    int numLabels = DRWN_INT_MAX);

int drwnConnectedComponents(cv::Mat& segments, bool b8Connected = false);
int drwnConnectedComponents(MatrixXi& segments, bool b8Connected = false);

cv::Mat drwnFastSuperpixels(const cv::Mat& img, unsigned gridSize);

cv::Mat drwnKMeansSegments(const cv::Mat& img, unsigned numCentroids);

cv::Mat drwnSLICSuperpixels(const cv::Mat& img, unsigned nClusters, 
    double spatialWeight = 200.0, double threshold = 1.0e-3);

void drwnMergeSuperpixels(const cv::Mat& img, cv::Mat& seg, unsigned maxSegs);

std::vector<std::pair<cv::Mat, unsigned> > drwnLoadCIFAR(const string& filename,
    unsigned headerBytes = 1, cv::Size sz = cv::Size(32, 32), unsigned nChannels = 3);