feature_detector.cpp

Go to the documentation of this file.

#include "ccny_rgbd/features/feature_detector.h"

namespace ccny_rgbd {

FeatureDetector::FeatureDetector():
  compute_descriptors_(false)
{

}

FeatureDetector::~FeatureDetector()
{

}

void FeatureDetector::findFeatures(RGBDFrame& frame)
{
  boost::mutex::scoped_lock(mutex_);
  
  const cv::Mat& input_img = frame.rgb_img;

  // convert from RGB to grayscale
  cv::Mat gray_img(input_img.rows, input_img.cols, CV_8UC1);
  cvtColor(input_img, gray_img, CV_BGR2GRAY);

  // blur if needed
  if(smooth_ > 0)
  {
    int blur_size = smooth_*2 + 1;
    cv::GaussianBlur(gray_img, gray_img, cv::Size(blur_size, blur_size), 0);
  }

  // find the 2D coordinates of keypoints
  findFeatures(frame, gray_img);

  // calculates the 3D position and covariance of features
  frame.computeDistributions(max_range_, max_stdev_);
}

void FeatureDetector::setSmooth(int smooth)
{
  smooth_ = smooth;
}

void FeatureDetector::setMaxRange(double max_range)
{
  max_range_ = max_range;
}

void FeatureDetector::setMaxStDev(double max_stdev)
{
  max_stdev_ = max_stdev;
}

int FeatureDetector::getSmooth() const
{
  return smooth_;
}

double FeatureDetector::getMaxRange() const
{
  return max_range_;
} 

double FeatureDetector::getMaxStDev() const
{
  return max_stdev_;
} 

} // namespace ccny_rgbd