MTF
Public Member Functions | Public Attributes | List of all members
utils::MTFNet Class Reference

Public Member Functions

 MTFNet (RegNetParams _rgparams)
 Constructs network and fills net_ object, sets mean_ , num_channels_, and input_geometry_.
 
void preprocessBatch (std::vector< cv::Mat > &batch)
 preprocess the data by subtracting the mean
 
void extractBatch (int bs, int bi, std::vector< cv::Mat > t_data, std::vector< cv::Mat > t_labels, std::vector< cv::Mat > &t_data_b, std::vector< cv::Mat > &t_labels_b)
 Extract minibatch.
 

Public Attributes

boost::shared_ptr< caffe::Net< float > > net_
 
boost::shared_ptr< caffe::Solver< float > > solver_
 
cv::Size input_geometry_
 
int num_channels_
 
cv::Scalar channel_mean_
 
RegNetParams rgparams
 

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