1 /* 2 * cv_image_process_helper.cpp - OpenCV image processing helpers functions 3 * 4 * Copyright (c) 2016-2017 Intel Corporation 5 * 6 * Licensed under the Apache License, Version 2.0 (the "License"); 7 * you may not use this file except in compliance with the License. 8 * You may obtain a copy of the License at 9 * 10 * http://www.apache.org/licenses/LICENSE-2.0 11 * 12 * Unless required by applicable law or agreed to in writing, software 13 * distributed under the License is distributed on an "AS IS" BASIS, 14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 * See the License for the specific language governing permissions and 16 * limitations under the License. 17 * 18 * Author: Andrey Parfenov <a1994ndrey (at) gmail.com> 19 * Author: Wind Yuan <feng.yuan (at) intel.com> 20 */ 21 22 #include "cv_image_process_helper.h" 23 24 namespace XCam { 25 26 27 CVImageProcessHelper::CVImageProcessHelper () 28 : CVBaseClass () 29 { 30 31 } 32 33 cv::Mat 34 CVImageProcessHelper::erosion (const cv::Mat &image, int erosion_size, int erosion_type) 35 { 36 cv::Mat element = cv::getStructuringElement (erosion_type, 37 cv::Size (2 * erosion_size + 1, 2 * erosion_size + 1), 38 cv::Point (erosion_size, erosion_size)); 39 cv::Mat eroded; 40 cv::erode (image, eroded, element); 41 return eroded.clone (); 42 } 43 44 float 45 CVImageProcessHelper::get_snr (const cv::Mat &noisy, const cv::Mat &noiseless) 46 { 47 cv::Mat temp_noisy, temp_noiseless; 48 noisy.convertTo (temp_noisy, CV_32FC1); 49 noiseless.convertTo (temp_noiseless, CV_32FC1); 50 cv::Mat numerator, denominator; 51 cv::pow (temp_noisy, 2, numerator); 52 cv::pow (temp_noisy - temp_noiseless, 2, denominator); 53 float res = cv::sum (numerator)[0] / cv::sum (denominator)[0]; 54 res = sqrt (res); 55 return res; 56 } 57 58 void 59 CVImageProcessHelper::compute_dft (const cv::Mat &image, cv::Mat &result) 60 { 61 cv::Mat padded; 62 int m = cv::getOptimalDFTSize (image.rows); 63 int n = cv::getOptimalDFTSize (image.cols); 64 cv::copyMakeBorder (image, padded, 0, m - image.rows, 0, n - image.cols, cv::BORDER_CONSTANT, cv::Scalar::all(0)); 65 cv::Mat planes[] = {cv::Mat_<float> (padded), cv::Mat::zeros (padded.size (), CV_32FC1)}; 66 cv::merge (planes, 2, result); 67 cv::dft (result, result); 68 } 69 70 void 71 CVImageProcessHelper::compute_idft (cv::Mat *input, cv::Mat &result) 72 { 73 cv::Mat fimg; 74 cv::merge (input, 2, fimg); 75 cv::idft (fimg, result, cv::DFT_REAL_OUTPUT + cv::DFT_SCALE); 76 } 77 78 void 79 CVImageProcessHelper::apply_constraints (cv::Mat &image, float threshold_min_value, float threshold_max_value, float min_value, float max_value) 80 { 81 for (int i = 0; i < image.rows; i++) { 82 for (int j = 0; j < image.cols; j++) { 83 if (image.at<float>(i, j) < threshold_min_value) 84 { 85 image.at<float>(i, j) = min_value; 86 } 87 if (image.at<float>(i, j) > threshold_max_value) 88 { 89 image.at<float>(i, j) = max_value; 90 } 91 } 92 } 93 } 94 95 void 96 CVImageProcessHelper::normalize_weights (cv::Mat &weights) 97 { 98 weights.convertTo (weights, CV_32FC1); 99 float sum = cv::sum (weights)[0]; 100 weights /= sum; 101 } 102 103 } 104