44 #include <pcl/recognition/quantizable_modality.h>
45 #include <pcl/recognition/region_xy.h>
46 #include <pcl/recognition/sparse_quantized_multi_mod_template.h>
94 initialize (
const std::size_t width,
const std::size_t height,
const std::size_t nr_bins)
96 maps_.resize(nr_bins,
nullptr);
101 const std::size_t mapsSize = width*height;
103 for (
auto &map : maps_)
106 map =
reinterpret_cast<unsigned char*
> (
aligned_malloc (mapsSize));
107 memset (map, 0, mapsSize);
115 for (
auto &map : maps_)
130 inline unsigned char &
131 operator() (
const std::size_t bin_index,
const std::size_t col_index,
const std::size_t row_index)
133 return (maps_[bin_index][row_index*width_ + col_index]);
140 inline unsigned char &
141 operator() (
const std::size_t bin_index,
const std::size_t index)
143 return (maps_[bin_index][index]);
149 inline unsigned char *
150 operator() (
const std::size_t bin_index)
152 return (maps_[bin_index]);
160 inline const unsigned char &
161 operator() (
const std::size_t bin_index,
const std::size_t col_index,
const std::size_t row_index)
const
163 return (maps_[bin_index][row_index*width_ + col_index]);
170 inline const unsigned char &
171 operator() (
const std::size_t bin_index,
const std::size_t index)
const
173 return (maps_[bin_index][index]);
179 inline const unsigned char *
180 operator() (
const std::size_t bin_index)
const
182 return (maps_[bin_index]);
191 std::size_t nr_bins_;
193 std::vector<unsigned char*> maps_;
203 LinearizedMaps () : width_ (0), height_ (0), mem_width_ (0), mem_height_ (0), step_size_ (0)
234 initialize (
const std::size_t width,
const std::size_t height,
const std::size_t step_size)
236 maps_.resize(step_size*step_size,
nullptr);
239 mem_width_ = width / step_size;
240 mem_height_ = height / step_size;
241 step_size_ = step_size;
243 const std::size_t mapsSize = mem_width_ * mem_height_;
245 for (
auto &map : maps_)
248 map =
reinterpret_cast<unsigned char*
> (
aligned_malloc (2*mapsSize));
249 memset (map, 0, 2*mapsSize);
257 for (
auto &map : maps_)
273 inline unsigned char *
274 operator() (
const std::size_t col_index,
const std::size_t row_index)
276 return (maps_[row_index*step_size_ + col_index]);
283 inline unsigned char *
284 getOffsetMap (
const std::size_t col_index,
const std::size_t row_index)
286 const std::size_t map_col = col_index % step_size_;
287 const std::size_t map_row = row_index % step_size_;
289 const std::size_t map_mem_col_index = col_index / step_size_;
290 const std::size_t map_mem_row_index = row_index / step_size_;
292 return (maps_[map_row*step_size_ + map_col] + map_mem_row_index*mem_width_ + map_mem_col_index);
301 std::size_t mem_width_;
303 std::size_t mem_height_;
305 std::size_t step_size_;
307 std::vector<unsigned char*> maps_;
350 const std::vector<MaskMap*> & masks,
365 std::vector<LINEMODDetection> & detections)
const;
377 std::vector<LINEMODDetection> & detections,
378 float min_scale = 0.6944444f,
379 float max_scale = 1.44f,
380 float scale_multiplier = 1.2f)
const;
388 std::vector<LINEMODDetection> & matches)
const;
396 template_threshold_ = threshold;
405 use_non_max_suppression_ = use_non_max_suppression;
414 average_detections_ = average_detections;
423 return (templates_[template_id]);
430 return (templates_.size ());
467 float template_threshold_;
469 bool use_non_max_suppression_;
471 bool average_detections_;
473 std::vector<SparseQuantizedMultiModTemplate> templates_;
Stores a set of energy maps.
std::size_t getNumOfBins() const
Returns the number of bins used for quantization (which is equal to the number of energy maps).
void releaseAll()
Releases the internal data.
void initialize(const std::size_t width, const std::size_t height, const std::size_t nr_bins)
Initializes the set of energy maps.
std::size_t getWidth() const
Returns the width of the energy maps.
virtual ~EnergyMaps()
Destructor.
std::size_t getHeight() const
Returns the height of the energy maps.
Template matching using the LINEMOD approach.
void deserialize(std::istream &stream)
Deserializes templates from the specified stream.
void saveTemplates(const char *file_name) const
Saves the stored templates to the specified file.
void detectTemplatesSemiScaleInvariant(const std::vector< QuantizableModality * > &modalities, std::vector< LINEMODDetection > &detections, float min_scale=0.6944444f, float max_scale=1.44f, float scale_multiplier=1.2f) const
Detects the stored templates in a semi scale invariant manner by applying the detection to multiple s...
const SparseQuantizedMultiModTemplate & getTemplate(int template_id) const
Returns the template with the specified ID.
void matchTemplates(const std::vector< QuantizableModality * > &modalities, std::vector< LINEMODDetection > &matches) const
Matches the stored templates to the supplied modality data.
std::size_t getNumOfTemplates() const
Returns the number of stored/trained templates.
void setDetectionThreshold(float threshold)
Sets the detection threshold.
void setDetectionAveraging(bool average_detections)
Enables/disables averaging of close detections.
virtual ~LINEMOD()
Destructor.
void serialize(std::ostream &stream) const
Serializes the stored templates to the specified stream.
void detectTemplates(const std::vector< QuantizableModality * > &modalities, std::vector< LINEMODDetection > &detections) const
Detects the stored templates in the supplied modality data.
void loadTemplates(std::vector< std::string > &file_names)
Loads templates from the specified files.
void setNonMaxSuppression(bool use_non_max_suppression)
Enables/disables non-maximum suppression.
int addTemplate(const SparseQuantizedMultiModTemplate &linemod_template)
Adds the specified template to the matching queue.
int createAndAddTemplate(const std::vector< QuantizableModality * > &modalities, const std::vector< MaskMap * > &masks, const RegionXY ®ion)
Creates a template from the specified data and adds it to the matching queue.
void loadTemplates(const char *file_name)
Loads templates from the specified file.
Stores a set of linearized maps.
std::size_t getHeight() const
Returns the height of the linearized map.
void initialize(const std::size_t width, const std::size_t height, const std::size_t step_size)
Initializes the linearized map.
LinearizedMaps()
Constructor.
unsigned char * getOffsetMap(const std::size_t col_index, const std::size_t row_index)
Returns a linearized map starting at the specified position.
void releaseAll()
Releases the internal memory.
std::size_t getStepSize() const
Returns the step-size used to construct the linearized map.
std::size_t getWidth() const
Returns the width of the linearized map.
std::size_t getMapMemorySize() const
Returns the size of the memory map.
virtual ~LinearizedMaps()
Destructor.
void aligned_free(void *ptr)
void * aligned_malloc(std::size_t size)
Defines all the PCL and non-PCL macros used.
Represents a detection of a template using the LINEMOD approach.
LINEMODDetection()
Constructor.
int template_id
ID of the detected template.
int y
y-position of the detection.
float scale
scale at which the template was detected.
float score
score of the detection.
int x
x-position of the detection.
Defines a region in XY-space.
A multi-modality template constructed from a set of quantized multi-modality features.