mirror of
https://github.com/2martens/uni.git
synced 2026-05-06 11:26:25 +02:00
[CCV] Added oriented and laplacian pyramid
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
57
ccv/saliency/includes/oriented_pyramid.h
Normal file
57
ccv/saliency/includes/oriented_pyramid.h
Normal file
@ -0,0 +1,57 @@
|
||||
#ifndef SHEET5_ORIENTED_PYRAMID_H
|
||||
#define SHEET5_ORIENTED_PYRAMID_H
|
||||
|
||||
#include <opencv2/opencv.hpp>
|
||||
#include "laplacian_pyramid.h"
|
||||
|
||||
class oriented_pyramid {
|
||||
private:
|
||||
std::vector<std::vector<cv::Mat>> _orientation_maps;
|
||||
std::vector<cv::Mat> _gabor_filters;
|
||||
std::vector<cv::Mat> _feature_maps;
|
||||
cv::Mat _C;
|
||||
|
||||
/**
|
||||
* Initializes the Gabor filters.
|
||||
* @param num_orientations the number of orientations to use
|
||||
*/
|
||||
void initialize_gabor_filters(float num_orientations);
|
||||
public:
|
||||
/**
|
||||
* Initializes the oriented pyramid.
|
||||
* @param pyramid the laplacian pyramid
|
||||
* @param num_orientations the number of Gabor filters to apply
|
||||
*/
|
||||
oriented_pyramid(const laplacian_pyramid& pyramid, int num_orientations);
|
||||
|
||||
/**
|
||||
* Computes the feature maps.
|
||||
*/
|
||||
void compute_feature_maps();
|
||||
|
||||
/**
|
||||
* Computes the conspicuity map.
|
||||
*/
|
||||
void compute_conspicuity_map();
|
||||
|
||||
/**
|
||||
* Returns the conspicuity map.
|
||||
* Has to be called after compute_conspicuity_map.
|
||||
*
|
||||
* @return conspicuity map
|
||||
*/
|
||||
cv::Mat get_conspicuity_map();
|
||||
|
||||
/**
|
||||
* Returns the feature map for the nth orientation.
|
||||
*
|
||||
* compute_feature_maps must be called first.
|
||||
*
|
||||
* @param orientation the nth orientation
|
||||
* @return the feature map
|
||||
*/
|
||||
cv::Mat get_feature_map(int orientation);
|
||||
};
|
||||
|
||||
|
||||
#endif //SHEET5_ORIENTED_PYRAMID_H
|
||||
Reference in New Issue
Block a user