2martens/uni
1
0
Fork 0
mirror of https://github.com/2martens/uni.git synced 2026-05-06 19:36:26 +02:00
Code Releases Activity

[CVV] Added orientation pyramid

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2017-05-12 16:11:31 +02:00
parent 59aef6a424
commit 8eb48b9b0a
4 changed files with 126 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
ccv/sheet5/oriented_pyramid.cpp Normal file
View File

@ -0,0 +1,59 @@
#include "oriented_pyramid.h"
oriented_pyramid::oriented_pyramid(const laplacian_pyramid &pyramid, int num_orientations) {
_gabor_filters = std::vector<cv::Mat>();
_orientation_maps = std::vector<std::vector<cv::Mat>>();
_feature_maps = std::vector<cv::Mat>();
initialize_gabor_filters(num_orientations);
unsigned long number_of_layers = pyramid.get_number_of_layers();
for (unsigned long i = 0; i < num_orientations; i++) {
std::vector<cv::Mat> orientation_vector = std::vector<cv::Mat>();
for (int layer = 0; layer < number_of_layers; layer++) {
cv::Mat filtered_image;
cv::filter2D(pyramid.get(layer), filtered_image, -1, _gabor_filters.at(i), cv::Point(-1, -1), 0, cv::BORDER_CONSTANT);
orientation_vector.push_back(filtered_image.clone());
}
_orientation_maps.push_back(orientation_vector);
}
}
void oriented_pyramid::initialize_gabor_filters(float num_orientations) {
cv::Size size = cv::Size(20, 20);
double wavelength = 3;
double standard_deviation = 18;
double start_level = num_orientations / 2.0;
for (double level = start_level; level >= 0; level--) {
_gabor_filters.push_back(cv::getGaborKernel(size, standard_deviation, (level/num_orientations) * CV_PI, wavelength,
1, 0, CV_32F));
}
double end_level = start_level;
start_level = num_orientations - 1;
for (double level = start_level; level > end_level; level--) {
_gabor_filters.push_back(cv::getGaborKernel(size, standard_deviation, (level/num_orientations) * CV_PI, wavelength,
1, 0, CV_32F));
}
}
void oriented_pyramid::compute_feature_maps() {
unsigned long num_orientations = _orientation_maps.size();
for (unsigned long i = 0; i < num_orientations; i++) {
cv::Mat feature_map = (cv::Mat &&) _orientation_maps.at(i).front();
cv::Size original_size = feature_map.size();
unsigned long num_layers = _orientation_maps.at(i).size();
for (unsigned long layer = 1; layer < num_layers; layer++) {
cv::Mat resized_image;
cv::resize(_orientation_maps.at(i).at(layer), resized_image, original_size, 0, 0, cv::INTER_CUBIC);
feature_map += resized_image;
}
_feature_maps.push_back(feature_map.clone());
}
}
cv::Mat oriented_pyramid::get_feature_map(int orientation) {
return _feature_maps.at((unsigned long) orientation);
}