rapid
A ROS robotics library.
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Pages
rapid::perception::ImageRecognizer Class Reference

#include <image_recognition.h>

Public Member Functions

 ImageRecognizer ()
 
 ImageRecognizer (boost::shared_ptr< caffe::Net< float > > feature_extraction_net, cv::Mat mean, cv::Size input_geometry, int num_channels, const std::vector< std::string > &labels)
 
void set_image (const cv::Mat &image)
 
cv::Mat layer (const std::string &layer_name, std::string *error)
 
std::vector< std::pair
< std::string, float > > 
predictions (int num_predictions, std::string *error)
 

Static Public Member Functions

static bool AlexNet (std::string model_dir, ImageRecognizer *recognizer, std::string *error)
 

Detailed Description

Definition at line 30 of file image_recognition.h.

Constructor & Destructor Documentation

rapid::perception::ImageRecognizer::ImageRecognizer ( )
rapid::perception::ImageRecognizer::ImageRecognizer ( boost::shared_ptr< caffe::Net< float > >  feature_extraction_net,
cv::Mat  mean,
cv::Size  input_geometry,
int  num_channels,
const std::vector< std::string > &  labels 
)

Member Function Documentation

static bool rapid::perception::ImageRecognizer::AlexNet ( std::string  model_dir,
ImageRecognizer recognizer,
std::string *  error 
)
static
cv::Mat rapid::perception::ImageRecognizer::layer ( const std::string &  layer_name,
std::string *  error 
)
std::vector<std::pair<std::string, float> > rapid::perception::ImageRecognizer::predictions ( int  num_predictions,
std::string *  error 
)
void rapid::perception::ImageRecognizer::set_image ( const cv::Mat &  image)

The documentation for this class was generated from the following file: