generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Changes done in npr module

Browse Source
This commit is contained in:
siddharth 2013-10-17 21:14:31 +05:30
parent 35d9ca6483
commit 2f65a0d4ba
4 changed files with 95 additions and 241 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

99
modules/photo/src/npr.cpp
View File

@ -79,34 +79,24 @@ void cv::detailEnhance(InputArray _src, OutputArray _dst, float sigma_s, float s
int h = I.size().height; int h = I.size().height;
int w = I.size().width; int w = I.size().width;
int channel = I.channels();
float factor = 3.0; float factor = 3.0;
Mat img = Mat(I.size(),CV_32FC3); Mat img = Mat(I.size(),CV_32FC3);
I.convertTo(img,CV_32FC3,1.0/255.0); I.convertTo(img,CV_32FC3,1.0/255.0);
Mat res = Mat(h,w,CV_32FC3); Mat res = Mat(h,w,CV_32FC1);
dst.convertTo(res,CV_32FC3,1.0/255.0); dst.convertTo(res,CV_32FC3,1.0/255.0);
Mat result = Mat(img.size(),CV_32FC3); Mat result = Mat(img.size(),CV_32FC3);
Mat lab = Mat(img.size(),CV_32FC3); Mat lab = Mat(img.size(),CV_32FC3);
Mat l_channel = Mat(img.size(),CV_32FC1); vector <Mat> lab_channel;
Mat a_channel = Mat(img.size(),CV_32FC1);
Mat b_channel = Mat(img.size(),CV_32FC1);
cvtColor(img,lab,COLOR_BGR2Lab); cvtColor(img,lab,COLOR_BGR2Lab);
split(lab,lab_channel);
for(int i = 0; i < h; i++)
for(int j = 0; j < w; j++)
{
l_channel.at<float>(i,j) = lab.at<float>(i,j*channel+0);
a_channel.at<float>(i,j) = lab.at<float>(i,j*channel+1);
b_channel.at<float>(i,j) = lab.at<float>(i,j*channel+2);
}
Mat L = Mat(img.size(),CV_32FC1); Mat L = Mat(img.size(),CV_32FC1);
l_channel.convertTo(L,CV_32FC1,1.0/255.0); lab_channel[0].convertTo(L,CV_32FC1,1.0/255.0);
Domain_Filter obj; Domain_Filter obj;
@ -114,23 +104,13 @@ void cv::detailEnhance(InputArray _src, OutputArray _dst, float sigma_s, float s
Mat detail = Mat(h,w,CV_32FC1); Mat detail = Mat(h,w,CV_32FC1);
for(int i = 0; i < h; i++) detail = L - res;
for(int j = 0; j < w; j++) multiply(detail,factor,detail);
detail.at<float>(i,j) = L.at<float>(i,j) - res.at<float>(i,j); L = res + detail;
for(int i = 0; i < h; i++) L.convertTo(lab_channel[0],CV_32FC1,255);
for(int j = 0; j < w; j++)
L.at<float>(i,j) = res.at<float>(i,j) + factor*detail.at<float>(i,j);
L.convertTo(l_channel,CV_32FC1,255); merge(lab_channel,lab);
for(int i = 0; i < h; i++)
for(int j = 0; j < w; j++)
{
lab.at<float>(i,j*channel+0) = l_channel.at<float>(i,j);
lab.at<float>(i,j*channel+1) = a_channel.at<float>(i,j);
lab.at<float>(i,j*channel+2) = b_channel.at<float>(i,j);
}
cvtColor(lab,result,COLOR_Lab2BGR); cvtColor(lab,result,COLOR_Lab2BGR);
result.convertTo(dst,CV_8UC3,255); result.convertTo(dst,CV_8UC3,255);
@ -174,64 +154,21 @@ void cv::stylization(InputArray _src, OutputArray _dst, float sigma_s, float sig
int channel = img.channels(); int channel = img.channels();
Mat res = Mat(h,w,CV_32FC3); Mat res = Mat(h,w,CV_32FC3);
Mat magnitude = Mat(h,w,CV_32FC1);
Domain_Filter obj; Domain_Filter obj;
obj.filter(img, res, sigma_s, sigma_r, NORMCONV_FILTER); obj.filter(img, res, sigma_s, sigma_r, NORMCONV_FILTER);
vector <Mat> planes; obj.find_magnitude(res,magnitude,2);
split(res, planes);
Mat magXR = Mat(h, w, CV_32FC1);
Mat magYR = Mat(h, w, CV_32FC1);
Mat magXG = Mat(h, w, CV_32FC1);
Mat magYG = Mat(h, w, CV_32FC1);
Mat magXB = Mat(h, w, CV_32FC1);
Mat magYB = Mat(h, w, CV_32FC1);
Sobel(planes[0], magXR, CV_32FC1, 1, 0, 3);
Sobel(planes[0], magYR, CV_32FC1, 0, 1, 3);
Sobel(planes[1], magXG, CV_32FC1, 1, 0, 3);
Sobel(planes[1], magYG, CV_32FC1, 0, 1, 3);
Sobel(planes[2], magXB, CV_32FC1, 1, 0, 3);
Sobel(planes[2], magYB, CV_32FC1, 0, 1, 3);
Mat magx = Mat(h,w,CV_32FC1);
Mat magy = Mat(h,w,CV_32FC1);
Mat mag1 = Mat(h,w,CV_32FC1);
Mat mag2 = Mat(h,w,CV_32FC1);
Mat mag3 = Mat(h,w,CV_32FC1);
magnitude(magXR,magYR,mag1);
magnitude(magXG,magYG,mag2);
magnitude(magXB,magYB,mag3);
Mat magnitude = Mat(h,w,CV_32FC1);
for(int i =0;i < h;i++)
for(int j=0;j<w;j++)
{
magnitude.at<float>(i,j) = mag1.at<float>(i,j) + mag2.at<float>(i,j) + mag3.at<float>(i,j);
}
for(int i =0;i < h;i++)
for(int j=0;j<w;j++)
{
magnitude.at<float>(i,j) = 1.0f - magnitude.at<float>(i,j);
}
Mat stylized = Mat(h,w,CV_32FC3); Mat stylized = Mat(h,w,CV_32FC3);
for(int i =0;i < h;i++) vector <Mat> temp;
for(int j=0;j<w;j++) split(res,temp);
for(int c=0;c<channel;c++) multiply(temp[0],magnitude,temp[0]);
{ multiply(temp[1],magnitude,temp[1]);
stylized.at<float>(i,j*channel + c) = res.at<float>(i,j*channel + c) * magnitude.at<float>(i,j); multiply(temp[2],magnitude,temp[2]);
} merge(temp,stylized);
stylized.convertTo(dst,CV_8UC3,255); stylized.convertTo(dst,CV_8UC3,255);
} }
@ -259,7 +196,7 @@ void cv::edgeEnhance(InputArray _src, OutputArray _dst, float sigma_s, float sig
obj.filter(img, res, sigma_s, sigma_r, NORMCONV_FILTER); obj.filter(img, res, sigma_s, sigma_r, NORMCONV_FILTER);
obj.find_magnitude(res,magnitude); obj.find_magnitude(res,magnitude,1);
magnitude.convertTo(dst,CV_8UC1,255); magnitude.convertTo(dst,CV_8UC1,255);
} }

162
modules/photo/src/npr.hpp
View File

@ -61,7 +61,7 @@ class Domain_Filter
void getGradienty( const Mat &img, Mat &gy); void getGradienty( const Mat &img, Mat &gy);
void diffx(const Mat &img, Mat &temp); void diffx(const Mat &img, Mat &temp);
void diffy(const Mat &img, Mat &temp); void diffy(const Mat &img, Mat &temp);
void find_magnitude(Mat &img, Mat &mag); void find_magnitude(Mat &img, Mat &mag, int flags);
void compute_boxfilter(Mat &output, Mat &hz, Mat &psketch, float radius); void compute_boxfilter(Mat &output, Mat &hz, Mat &psketch, float radius);
void compute_Rfilter(Mat &O, Mat &horiz, float sigma_h); void compute_Rfilter(Mat &O, Mat &horiz, float sigma_h);
void compute_NCfilter(Mat &O, Mat &horiz, Mat &psketch, float radius); void compute_NCfilter(Mat &O, Mat &horiz, Mat &psketch, float radius);
@ -131,9 +131,8 @@ void Domain_Filter::getGradienty( const Mat &img, Mat &gy)
} }
} }
void Domain_Filter::find_magnitude(Mat &img, Mat &mag) void Domain_Filter::find_magnitude(Mat &img, Mat &mag, int flags)
{ {
int h = img.rows; int h = img.rows;
int w = img.cols; int w = img.cols;
@ -149,17 +148,28 @@ void Domain_Filter::find_magnitude(Mat &img, Mat &mag)
Mat magXB = Mat(h, w, CV_32FC1); Mat magXB = Mat(h, w, CV_32FC1);
Mat magYB = Mat(h, w, CV_32FC1); Mat magYB = Mat(h, w, CV_32FC1);
getGradientx(planes[0], magXR); if(flags == 1)
getGradienty(planes[0], magYR); {
getGradientx(planes[0], magXR);
getGradienty(planes[0], magYR);
getGradientx(planes[1], magXG); getGradientx(planes[1], magXG);
getGradienty(planes[1], magYG); getGradienty(planes[1], magYG);
getGradientx(planes[2], magXR); getGradientx(planes[2], magXR);
getGradienty(planes[2], magYR); getGradienty(planes[2], magYR);
}
else if(flags == 2)
{
Sobel(planes[0], magXR, CV_32FC1, 1, 0, 3);
Sobel(planes[0], magYR, CV_32FC1, 0, 1, 3);
Mat magx = Mat(h,w,CV_32FC1); Sobel(planes[1], magXG, CV_32FC1, 1, 0, 3);
Mat magy = Mat(h,w,CV_32FC1); Sobel(planes[1], magYG, CV_32FC1, 0, 1, 3);
Sobel(planes[2], magXB, CV_32FC1, 1, 0, 3);
Sobel(planes[2], magYB, CV_32FC1, 0, 1, 3);
}
Mat mag1 = Mat(h,w,CV_32FC1); Mat mag1 = Mat(h,w,CV_32FC1);
Mat mag2 = Mat(h,w,CV_32FC1); Mat mag2 = Mat(h,w,CV_32FC1);
@ -169,39 +179,21 @@ void Domain_Filter::find_magnitude(Mat &img, Mat &mag)
magnitude(magXG,magYG,mag2); magnitude(magXG,magYG,mag2);
magnitude(magXB,magYB,mag3); magnitude(magXB,magYB,mag3);
for(int i =0;i < h;i++) mag = mag1 + mag2 + mag3;
for(int j=0;j<w;j++) mag = 1.0f - mag;
{
mag.at<float>(i,j) = mag1.at<float>(i,j) + mag2.at<float>(i,j) + mag3.at<float>(i,j);
}
for(int i =0;i < h;i++)
for(int j=0;j<w;j++)
{
mag.at<float>(i,j) = 1.0f - mag.at<float>(i,j);
}
} }
void Domain_Filter::compute_Rfilter(Mat &output, Mat &hz, float sigma_h) void Domain_Filter::compute_Rfilter(Mat &output, Mat &hz, float sigma_h)
{ {
float a;
int h = output.rows; int h = output.rows;
int w = output.cols; int w = output.cols;
int channel = output.channels(); int channel = output.channels();
a = (float) exp((-1.0 * sqrt(2.0)) / sigma_h); float a = (float) exp((-1.0 * sqrt(2.0)) / sigma_h);
Mat temp = Mat(h,w,CV_32FC3); Mat temp = Mat(h,w,CV_32FC3);
for(int i =0; i < h;i++) output.copyTo(temp);
for(int j=0;j<w;j++)
for(int c=0;c<channel;c++)
temp.at<float>(i,j*channel+c) = output.at<float>(i,j*channel+c);
Mat V = Mat(h,w,CV_32FC1); Mat V = Mat(h,w,CV_32FC1);
for(int i=0;i<h;i++) for(int i=0;i<h;i++)
@ -232,14 +224,7 @@ void Domain_Filter::compute_Rfilter(Mat &output, Mat &hz, float sigma_h)
} }
} }
temp.copyTo(output);
for(int i =0; i < h;i++)
for(int j=0;j<w;j++)
for(int c=0;c<channel;c++)
output.at<float>(i,j*channel+c) = temp.at<float>(i,j*channel+c);
temp.release();
V.release();
} }
void Domain_Filter::compute_boxfilter(Mat &output, Mat &hz, Mat &psketch, float radius) void Domain_Filter::compute_boxfilter(Mat &output, Mat &hz, Mat &psketch, float radius)
@ -249,12 +234,8 @@ void Domain_Filter::compute_boxfilter(Mat &output, Mat &hz, Mat &psketch, float
Mat lower_pos = Mat(h,w,CV_32FC1); Mat lower_pos = Mat(h,w,CV_32FC1);
Mat upper_pos = Mat(h,w,CV_32FC1); Mat upper_pos = Mat(h,w,CV_32FC1);
for(int i=0;i<h;i++) lower_pos = hz - radius;
for(int j=0;j<w;j++) upper_pos = hz + radius;
{
lower_pos.at<float>(i,j) = hz.at<float>(i,j) - radius;
upper_pos.at<float>(i,j) = hz.at<float>(i,j) + radius;
}
lower_idx = Mat::zeros(h,w,CV_32FC1); lower_idx = Mat::zeros(h,w,CV_32FC1);
upper_idx = Mat::zeros(h,w,CV_32FC1); upper_idx = Mat::zeros(h,w,CV_32FC1);
@ -334,19 +315,11 @@ void Domain_Filter::compute_boxfilter(Mat &output, Mat &hz, Mat &psketch, float
upper_idx.at<float>(i,j) = temp_upper_idx.at<float>(0,j) + 1; upper_idx.at<float>(i,j) = temp_upper_idx.at<float>(0,j) + 1;
} }
lower_pos_row.release();
upper_pos_row.release();
temp_lower_idx.release();
temp_upper_idx.release();
} }
for(int i=0;i<h;i++) psketch = upper_idx - lower_idx;
for(int j=0;j<w;j++)
psketch.at<float>(i,j) = upper_idx.at<float>(i,j) - lower_idx.at<float>(i,j);
} }
void Domain_Filter::compute_NCfilter(Mat &output, Mat &hz, Mat &psketch, float radius) void Domain_Filter::compute_NCfilter(Mat &output, Mat &hz, Mat &psketch, float radius)
{ {
int h = output.rows; int h = output.rows;
int w = output.cols; int w = output.cols;
int channel = output.channels(); int channel = output.channels();
@ -377,24 +350,25 @@ void Domain_Filter::compute_NCfilter(Mat &output, Mat &hz, Mat &psketch, float r
Mat a = Mat::zeros(h,w,CV_32FC1); Mat a = Mat::zeros(h,w,CV_32FC1);
Mat b = Mat::zeros(h,w,CV_32FC1); Mat b = Mat::zeros(h,w,CV_32FC1);
// Compute the box filter using a summed area table.
for(int c=0;c<channel;c++) for(int c=0;c<channel;c++)
{ {
Mat flag = Mat::ones(h,w,CV_32FC1); Mat flag = Mat::ones(h,w,CV_32FC1);
for(int i=0;i<h;i++) multiply(flag,c+1,flag);
for(int j=0;j<w;j++)
flag.at<float>(i,j) = (c+1)*flag.at<float>(i,j);
for(int i=0;i<h;i++) Mat temp1, temp2;
for(int j=0;j<w;j++) multiply(flag - 1,h*(w+1),temp1);
{ multiply(lower_idx - 1,h,temp2);
a.at<float>(i,j) = (flag.at<float>(i,j) - 1) * h * (w+1) + (lower_idx.at<float>(i,j) - 1) * h + indices.at<float>(i,j); a = temp1 + temp2 + indices;
b.at<float>(i,j) = (flag.at<float>(i,j) - 1) * h * (w+1) + (upper_idx.at<float>(i,j) - 1) * h + indices.at<float>(i,j);
} multiply(flag - 1,h*(w+1),temp1);
multiply(upper_idx - 1,h,temp2);
b = temp1 + temp2 + indices;
int p,q,r,rem; int p,q,r,rem;
int p1,q1,r1,rem1; int p1,q1,r1,rem1;
// Calculating indices
for(int i=0;i<h;i++) for(int i=0;i<h;i++)
{ {
for(int j=0;j<w;j++) for(int j=0;j<w;j++)
@ -416,7 +390,6 @@ void Domain_Filter::compute_NCfilter(Mat &output, Mat &hz, Mat &psketch, float r
q=q-1; q=q-1;
} }
r1 = (int) a.at<float>(i,j)/(h*(w+1)); r1 = (int) a.at<float>(i,j)/(h*(w+1));
rem1 = (int) a.at<float>(i,j) - r1*h*(w+1); rem1 = (int) a.at<float>(i,j) - r1*h*(w+1);
q1 = rem1/h; q1 = rem1/h;
@ -427,18 +400,13 @@ void Domain_Filter::compute_NCfilter(Mat &output, Mat &hz, Mat &psketch, float r
q1=q1-1; q1=q1-1;
} }
final.at<float>(i,j*channel+2-c) = (box_filter.at<float>(p-1,q*channel+(2-r)) - box_filter.at<float>(p1-1,q1*channel+(2-r1))) final.at<float>(i,j*channel+2-c) = (box_filter.at<float>(p-1,q*channel+(2-r)) - box_filter.at<float>(p1-1,q1*channel+(2-r1)))
/(upper_idx.at<float>(i,j) - lower_idx.at<float>(i,j)); /(upper_idx.at<float>(i,j) - lower_idx.at<float>(i,j));
} }
} }
} }
for(int i=0;i<h;i++) final.copyTo(output);
for(int j=0;j<w;j++)
for(int c=0;c<channel;c++)
output.at<float>(i,j*channel+c) = final.at<float>(i,j*channel+c);
} }
void Domain_Filter::init(const Mat &img, int flags, float sigma_s, float sigma_r) void Domain_Filter::init(const Mat &img, int flags, float sigma_s, float sigma_r)
{ {
@ -482,20 +450,15 @@ void Domain_Filter::init(const Mat &img, int flags, float sigma_s, float sigma_r
Mat final = Mat(h,w,CV_32FC3); Mat final = Mat(h,w,CV_32FC3);
for(int i = 0; i < h; i++) Mat tempx,tempy;
for(int j = 0; j < w; j++) multiply(distx,sigma_s/sigma_r,tempx);
{ multiply(disty,sigma_s/sigma_r,tempy);
horiz.at<float>(i,j) = (float) 1.0 + (sigma_s/sigma_r) * distx.at<float>(i,j);
vert.at<float>(i,j) = (float) 1.0 + (sigma_s/sigma_r) * disty.at<float>(i,j);
}
horiz = 1.0f + tempx;
vert = 1.0f + tempy;
O = Mat(h,w,CV_32FC3); O = Mat(h,w,CV_32FC3);
img.copyTo(O);
for(int i =0;i<h;i++)
for(int j =0;j<w;j++)
for(int c=0;c<channel;c++)
O.at<float>(i,j*channel+c) = img.at<float>(i,j*channel+c);
O_t = Mat(w,h,CV_32FC3); O_t = Mat(w,h,CV_32FC3);
@ -588,17 +551,14 @@ void Domain_Filter::pencil_sketch(const Mat &img, Mat &sketch, Mat &color_res, f
init(img,2,sigma_s,sigma_r); init(img,2,sigma_s,sigma_r);
int h = img.size().height; int h = img.size().height;
int w = img.size().width; int w = img.size().width;
int channel = img.channels();
/////////////////////// convert to YCBCR model for color pencil drawing ////////////////////////////////////////////////////// /////////////////////// convert to YCBCR model for color pencil drawing //////////////////////////////////////////////////////
Mat color_sketch = Mat(h,w,CV_32FC3); Mat color_sketch = Mat(h,w,CV_32FC3);
Mat Y_channel = Mat(h,w,CV_32FC1);
Mat U_channel = Mat(h,w,CV_32FC1);
Mat V_channel = Mat(h,w,CV_32FC1);
cvtColor(img,color_sketch,COLOR_BGR2YCrCb); cvtColor(img,color_sketch,COLOR_BGR2YCrCb);
vector <Mat> YUV_channel;
Mat vert_t = ct_V.t(); Mat vert_t = ct_V.t();
float sigma_h = sigma_s; float sigma_h = sigma_s;
@ -635,33 +595,11 @@ void Domain_Filter::pencil_sketch(const Mat &img, Mat &sketch, Mat &color_res, f
if(i==0) if(i==0)
{ {
sketch = pen_res.clone(); sketch = pen_res.clone();
split(color_sketch,YUV_channel);
for(int k = 0; k < h; k++) pen_res.copyTo(YUV_channel[0]);
for(int j = 0; j < w; j++) merge(YUV_channel,color_sketch);
{
Y_channel.at<float>(k,j) = color_sketch.at<float>(k,j*channel+0);
U_channel.at<float>(k,j) = color_sketch.at<float>(k,j*channel+1);
V_channel.at<float>(k,j) = color_sketch.at<float>(k,j*channel+2);
}
for(int k=0;k<h;k++)
for(int j=0;j<w;j++)
Y_channel.at<float>(k,j) = pen_res.at<float>(k,j);
// cvMerge(Y_channel,U_channel,V_channel,0,color_sketch);
for(int k = 0; k < h; k++)
for(int j = 0; j < w; j++)
{
color_sketch.at<float>(k,j*channel+0) = Y_channel.at<float>(k,j);
color_sketch.at<float>(k,j*channel+1) = U_channel.at<float>(k,j);
color_sketch.at<float>(k,j*channel+2) = V_channel.at<float>(k,j);
}
cvtColor(color_sketch,color_res,COLOR_YCrCb2BGR); cvtColor(color_sketch,color_res,COLOR_YCrCb2BGR);
} }
} }
} }

30
modules/photo/src/seamless_cloning.cpp
View File

@ -41,7 +41,6 @@
#include "precomp.hpp" #include "precomp.hpp"
#include "opencv2/photo.hpp" #include "opencv2/photo.hpp"
#include "opencv2/highgui.hpp"
#include <iostream> #include <iostream>
#include <stdlib.h> #include <stdlib.h>
@ -100,30 +99,17 @@ void cv::seamlessClone(InputArray _src, InputArray _dst, InputArray _mask, Point
exit(0); exit(0);
} }
for(int i=minx, k=minxd;i<(minx+lenx);i++,k++) Rect roi_d(minyd,minxd,leny,lenx);
for(int j=miny,l=minyd ;j<(miny+leny);j++,l++) Rect roi_s(miny,minx,leny,lenx);
{
dst_mask.at<uchar>(k,l) = gray.at<uchar>(i,j);
}
int channel = 3; Mat destinationROI = dst_mask(roi_d);
Mat sourceROI = cs_mask(roi_s);
for(int i=minx;i<(minx+lenx);i++) gray(roi_s).copyTo(destinationROI);
for(int j=miny;j<(miny+leny);j++) src(roi_s).copyTo(sourceROI,gray(roi_s));
{
for(int c=0;c<3;c++)
{
if(gray.at<uchar>(i,j) == 255)
cs_mask.at<uchar>(i,j*channel+c) = src.at<uchar>(i,j*channel+c);
}
}
for(int i=minx, k=minxd;i<(minx+lenx);i++,k++) destinationROI = cd_mask(roi_d);
for(int j=miny,l=minyd ;j<(miny+leny);j++,l++) cs_mask(roi_s).copyTo(destinationROI);
{
for(int c=0;c<channel;c++)
cd_mask.at<uchar>(k,l*channel+c) = cs_mask.at<uchar>(i,j*channel+c);
}
Cloning obj; Cloning obj;
obj.normal_clone(dest,cd_mask,dst_mask,blend,flags); obj.normal_clone(dest,cd_mask,dst_mask,blend,flags);

45
modules/photo/src/seamless_cloning.hpp
View File

@ -555,8 +555,6 @@ void Cloning::local_color_change(Mat &I, Mat &mask, Mat &wmask, Mat &cloned, flo
void Cloning::illum_change(Mat &I, Mat &mask, Mat &wmask, Mat &cloned, float alpha, float beta) void Cloning::illum_change(Mat &I, Mat &mask, Mat &wmask, Mat &cloned, float alpha, float beta)
{ {
int channel = I.channels();
initialization(I,mask,wmask); initialization(I,mask,wmask);
array_product(srx32,sgx,smask); array_product(srx32,sgx,smask);
@ -565,18 +563,20 @@ void Cloning::illum_change(Mat &I, Mat &mask, Mat &wmask, Mat &cloned, float alp
Mat mag = Mat(I.size(),CV_32FC3); Mat mag = Mat(I.size(),CV_32FC3);
magnitude(srx32,sry32,mag); magnitude(srx32,sry32,mag);
for(int i=0;i < I.size().height; i++) Mat multX, multY, multx_temp, multy_temp;
for(int j=0; j < I.size().width; j++)
for(int c=0;c < channel;++c) multiply(srx32,pow(alpha,beta),multX);
{ pow(mag,-1*beta, multx_temp);
if(srx32.at<float>(i,j*channel+c) != 0) multiply(multX,multx_temp,srx32);
{
srx32.at<float>(i,j*channel+c) = multiply(sry32,pow(alpha,beta),multY);
pow(alpha,beta)*srx32.at<float>(i,j*channel+c)*pow(mag.at<float>(i,j*channel+c),-1*beta); pow(mag,-1*beta, multy_temp);
sry32.at<float>(i,j*channel+c) = multiply(multY,multy_temp,sry32);
pow(alpha,beta)*sry32.at<float>(i,j*channel+c)*pow(mag.at<float>(i,j*channel+c),-1*beta);
} Mat zeroMask = (srx32 != 0);
}
srx32.copyTo(srx32, zeroMask);
sry32.copyTo(sry32, zeroMask);
evaluate(I,wmask,cloned); evaluate(I,wmask,cloned);
} }
@ -584,23 +584,16 @@ void Cloning::illum_change(Mat &I, Mat &mask, Mat &wmask, Mat &cloned, float alp
void Cloning::texture_flatten(Mat &I, Mat &mask, Mat &wmask, double low_threshold, void Cloning::texture_flatten(Mat &I, Mat &mask, Mat &wmask, double low_threshold,
double high_threshold, int kernel_size, Mat &cloned) double high_threshold, int kernel_size, Mat &cloned)
{ {
int channel = mask.channels();
initialization(I,mask,wmask); initialization(I,mask,wmask);
Mat out = Mat(mask.size(),CV_8UC1); Mat out = Mat(mask.size(),CV_8UC1);
Canny(mask,out,low_threshold,high_threshold,kernel_size); Canny(mask,out,low_threshold,high_threshold,kernel_size);
for(int i=0;i<mask.size().height;i++) Mat zeros(sgx.size(), CV_32FC3);
for(int j=0;j<mask.size().width;j++) zeros.setTo(0);
for(int c=0;c<channel;c++) Mat zerosMask = (out != 255);
{ zeros.copyTo(sgx, zerosMask);
if(out.at<uchar>(i,j) != 255) zeros.copyTo(sgy, zerosMask);
{
sgx.at<float>(i,j*channel+c) = 0.0;
sgy.at<float>(i,j*channel+c) = 0.0;
}
}
array_product(srx32,sgx,smask); array_product(srx32,sgx,smask);
array_product(sry32,sgy,smask); array_product(sry32,sgy,smask);