Skip to content

O
opencv
Commit 5b0ee9e0 authored by siddharth's avatar siddharth
Browse files
Options

decolor module updated

parent 0802c065
Hide whitespace changes
Inline Side-by-side
Showing with 422 additions and 367 deletions
modules/photo/src/contrast_preserve.cpp
View file @ 5b0ee9e0
#include "precomp.hpp" #include "precomp.hpp"
#include "opencv2/photo.hpp" #include "opencv2/photo.hpp"
#include "opencv2/imgproc.hpp" #include "opencv2/imgproc.hpp"
#include <opencv2/highgui.hpp>
#include "math.h" #include "math.h"
#include <vector> #include <vector>
#include <limits> #include <limits>
...@@ -10,11 +11,11 @@ ...@@ -10,11 +11,11 @@
using namespace std; using namespace std;
using namespace cv; using namespace cv;
int rounding(double a); double norm(double);
int rounding(double a) double norm(double E)
{ {
return int(a + 0.5); return (sqrt(pow(E,2)));
} }
void cv::decolor(InputArray _src, OutputArray _dst, OutputArray _boost) void cv::decolor(InputArray _src, OutputArray _dst, OutputArray _boost)
...@@ -37,33 +38,20 @@ void cv::decolor(InputArray _src, OutputArray _dst, OutputArray _boost) ...@@ -37,33 +38,20 @@ void cv::decolor(InputArray _src, OutputArray _dst, OutputArray _boost)
return; return;
} }
float sigma = .02; int maxIter = 15;
int maxIter = 8;
int iterCount = 0; int iterCount = 0;
float tol = .0001;
int h = I.size().height; double E = 0;
int w = I.size().width; double pre_E = std::numeric_limits<double>::infinity();
Decolor obj; Decolor obj;
Mat img; Mat img;
double sizefactor;
if((h + w) > 900) img = Mat(I.size(),CV_32FC3);
{ I.convertTo(img,CV_32FC3,1.0/255.0);
sizefactor = (double)900/(h+w);
resize(I,I,Size(rounding(h*sizefactor),rounding(w*sizefactor)));
img = Mat(I.size(),CV_32FC3);
I.convertTo(img,CV_32FC3,1.0/255.0);
}
else
{
img = Mat(I.size(),CV_32FC3);
I.convertTo(img,CV_32FC3,1.0/255.0);
}
obj.init(); obj.init();
vector <double> Cg; vector <double> Cg;
vector < vector <double> > polyGrad; vector < vector <double> > polyGrad;
...@@ -83,9 +71,10 @@ void cv::decolor(InputArray _src, OutputArray _dst, OutputArray _boost) ...@@ -83,9 +71,10 @@ void cv::decolor(InputArray _src, OutputArray _dst, OutputArray _boost)
//////////////////////////////// main loop starting //////////////////////////////////////// //////////////////////////////// main loop starting ////////////////////////////////////////
while (iterCount < maxIter) while(norm(E-pre_E) > tol)
{ {
iterCount +=1; iterCount +=1;
pre_E = E;
vector <double> G_pos; vector <double> G_pos;
vector <double> G_neg; vector <double> G_neg;
...@@ -150,6 +139,11 @@ void cv::decolor(InputArray _src, OutputArray _dst, OutputArray _boost) ...@@ -150,6 +139,11 @@ void cv::decolor(InputArray _src, OutputArray _dst, OutputArray _boost)
for(unsigned int i =0;i<wei.size();i++) for(unsigned int i =0;i<wei.size();i++)
wei[i] = wei1[i]; wei[i] = wei1[i];
E = obj.energyCalcu(Cg,polyGrad,wei);
if(iterCount > maxIter)
break;
G_pos.clear(); G_pos.clear();
G_neg.clear(); G_neg.clear();
temp.clear(); temp.clear();
......
modules/photo/src/contrast_preserve.hpp
View file @ 5b0ee9e0
...@@ -10,411 +10,472 @@ using namespace cv; ...@@ -10,411 +10,472 @@ using namespace cv;
class Decolor class Decolor
{ {
private: private:
Mat kernel; Mat kernel;
Mat kernel1; Mat kernel1;
int order; int order;
public: public:
void init(); void init();
vector<double> product(vector < vector<int> > &comb, vector <double> &initRGB); vector<double> product(vector < vector<int> > &comb, vector <double> &initRGB);
void singleChannelGradx(const Mat &img, Mat& dest); double energyCalcu(vector <double> &Cg, vector < vector <double> > &polyGrad, vector <double> &wei);
void singleChannelGrady(const Mat &img, Mat& dest); void singleChannelGradx(const Mat &img, Mat& dest);
void gradvector(const Mat &img, vector <double> &grad); void singleChannelGrady(const Mat &img, Mat& dest);
void colorGrad(Mat img, vector <double> &Cg); void gradvector(const Mat &img, vector <double> &grad);
void add_vector(vector < vector <int> > &comb, int r,int g,int b); void colorGrad(Mat img, vector <double> &Cg);
void add_to_vector_poly(vector < vector <double> > &polyGrad, vector <double> &curGrad); void add_vector(vector < vector <int> > &comb, int r,int g,int b);
void weak_order(Mat img, vector <double> &alf); void add_to_vector_poly(vector < vector <double> > &polyGrad, vector <double> &curGrad);
void grad_system(Mat img, vector < vector < double > > &polyGrad, vector < double > &Cg, vector < vector <int> >& comb); void weak_order(Mat img, vector <double> &alf);
void wei_update_matrix(vector < vector <double> > &poly, vector <double> &Cg, Mat &X); void grad_system(Mat img, vector < vector < double > > &polyGrad,
void wei_inti(vector < vector <int> > &comb, vector <double> &wei); vector < double > &Cg, vector < vector <int> >& comb);
void grayImContruct(vector <double> &wei, Mat img, Mat &Gray); void wei_update_matrix(vector < vector <double> > &poly, vector <double> &Cg, Mat &X);
void wei_inti(vector < vector <int> > &comb, vector <double> &wei);
void grayImContruct(vector <double> &wei, Mat img, Mat &Gray);
}; };
int rounding(double a);
int rounding(double a)
{
return int(a + 0.5);
}
float sigma = .02;
double Decolor::energyCalcu(vector <double> &Cg, vector < vector <double> > &polyGrad, vector <double> &wei)
{
vector <double> P;
vector <double> temp;
vector <double> temp1;
double val = 0.0;
for(unsigned int i=0;i< polyGrad[0].size();i++)
{
val = 0.0;
for(unsigned int j =0;j<polyGrad.size();j++)
val = val + (polyGrad[j][i] * wei[j]);
temp.push_back(val - Cg[i]);
temp1.push_back(val + Cg[i]);
}
for(unsigned int i=0;i<polyGrad[0].size();i++)
P.push_back(-1.0*log(exp(-1.0*pow(temp[i],2)/sigma) + exp(-1.0*pow(temp1[i],2)/sigma)));
double sum = 0.0;
for(unsigned int i=0;i<polyGrad[0].size();i++)
sum +=P[i];
return (sum/polyGrad[0].size());
}
void Decolor::init() void Decolor::init()
{ {
kernel = Mat(1,2, CV_32FC1); kernel = Mat(1,2, CV_32FC1);
kernel1 = Mat(2,1, CV_32FC1); kernel1 = Mat(2,1, CV_32FC1);
kernel.at<float>(0,0)=1.0; kernel.at<float>(0,0)=1.0;
kernel.at<float>(0,1)=-1.0; kernel.at<float>(0,1)=-1.0;
kernel1.at<float>(0,0)=1.0; kernel1.at<float>(0,0)=1.0;
kernel1.at<float>(1,0)=-1.0; kernel1.at<float>(1,0)=-1.0;
order = 2; order = 2;
} }
vector<double> Decolor::product(vector < vector<int> > &comb, vector <double> &initRGB) vector<double> Decolor::product(vector < vector<int> > &comb, vector <double> &initRGB)
{ {
vector <double> res; vector <double> res;
float dp; float dp;
for (unsigned int i=0;i<comb.size();i++) for (unsigned int i=0;i<comb.size();i++)
{ {
dp = 0.0f; dp = 0.0f;
for(int j=0;j<3;j++) for(int j=0;j<3;j++)
dp += comb[i][j] * initRGB[j]; dp += comb[i][j] * initRGB[j];
res.push_back(dp); res.push_back(dp);
} }
return res; return res;
} }
void Decolor::singleChannelGradx(const Mat &img, Mat& dest) void Decolor::singleChannelGradx(const Mat &img, Mat& dest)
{ {
int w=img.size().width; int w=img.size().width;
int h=img.size().height; int h=img.size().height;
Point anchor(kernel.cols - kernel.cols/2 - 1, kernel.rows - kernel.rows/2 - 1); Point anchor(kernel.cols - kernel.cols/2 - 1, kernel.rows - kernel.rows/2 - 1);
filter2D(img, dest, -1, kernel, anchor, 0.0, BORDER_CONSTANT); filter2D(img, dest, -1, kernel, anchor, 0.0, BORDER_CONSTANT);
for(int i=0;i<h;i++) for(int i=0;i<h;i++)
dest.at<float>(i,w-1)=0.0; dest.at<float>(i,w-1)=0.0;
} }
void Decolor::singleChannelGrady(const Mat &img, Mat& dest) void Decolor::singleChannelGrady(const Mat &img, Mat& dest)
{ {
int w=img.size().width; int w=img.size().width;
int h=img.size().height; int h=img.size().height;
Point anchor(kernel1.cols - kernel1.cols/2 - 1, kernel1.rows - kernel1.rows/2 - 1); Point anchor(kernel1.cols - kernel1.cols/2 - 1, kernel1.rows - kernel1.rows/2 - 1);
filter2D(img, dest, -1, kernel1, anchor, 0.0, BORDER_CONSTANT); filter2D(img, dest, -1, kernel1, anchor, 0.0, BORDER_CONSTANT);
for(int j=0;j<w;j++) for(int j=0;j<w;j++)
dest.at<float>(h-1,j)=0.0; dest.at<float>(h-1,j)=0.0;
} }
void Decolor::gradvector(const Mat &img, vector <double> &grad) void Decolor::gradvector(const Mat &img, vector <double> &grad)
{ {
Mat dest= Mat(img.size().height,img.size().width, CV_32FC1); Mat dest= Mat(img.size().height,img.size().width, CV_32FC1);
Mat dest1= Mat(img.size().height,img.size().width, CV_32FC1); Mat dest1= Mat(img.size().height,img.size().width, CV_32FC1);
singleChannelGradx(img,dest); singleChannelGradx(img,dest);
singleChannelGrady(img,dest1); singleChannelGrady(img,dest1);
Mat d_trans=dest.t(); Mat d_trans=dest.t();
Mat d1_trans=dest1.t(); Mat d1_trans=dest1.t();
int height = d_trans.size().height; int height = d_trans.size().height;
int width = d_trans.size().width; int width = d_trans.size().width;
for(int i=0;i<height;i++) for(int i=0;i<height;i++)
for(int j=0;j<width;j++) for(int j=0;j<width;j++)
grad.push_back(d_trans.at<float>(i,j)); grad.push_back(d_trans.at<float>(i,j));
for(int i=0;i<height;i++) for(int i=0;i<height;i++)
for(int j=0;j<width;j++) for(int j=0;j<width;j++)
grad.push_back(d1_trans.at<float>(i,j)); grad.push_back(d1_trans.at<float>(i,j));
dest.release(); dest.release();
dest1.release(); dest1.release();
} }
void Decolor::colorGrad(Mat img, vector <double> &Cg) void Decolor::colorGrad(Mat img, vector <double> &Cg)
{ {
Mat lab = Mat(img.size(),CV_32FC3); Mat lab = Mat(img.size(),CV_32FC3);
Mat l_channel = Mat(img.size(),CV_32FC1); Mat l_channel = Mat(img.size(),CV_32FC1);
Mat a_channel = Mat(img.size(),CV_32FC1); Mat a_channel = Mat(img.size(),CV_32FC1);
Mat b_channel = Mat(img.size(),CV_32FC1); Mat b_channel = Mat(img.size(),CV_32FC1);
cvtColor(img,lab,COLOR_BGR2Lab); cvtColor(img,lab,COLOR_BGR2Lab);
for(int i=0;i<img.size().height;i++) for(int i=0;i<img.size().height;i++)
for(int j=0;j<img.size().width;j++) for(int j=0;j<img.size().width;j++)
{ {
l_channel.at<float>(i,j) = lab.at<float>(i,j*3+0); l_channel.at<float>(i,j) = lab.at<float>(i,j*3+0);
a_channel.at<float>(i,j) = lab.at<float>(i,j*3+1); a_channel.at<float>(i,j) = lab.at<float>(i,j*3+1);
b_channel.at<float>(i,j) = lab.at<float>(i,j*3+2); b_channel.at<float>(i,j) = lab.at<float>(i,j*3+2);
} }
vector <double> ImL; vector <double> ImL;
vector <double> Ima; vector <double> Ima;
vector <double> Imb; vector <double> Imb;
gradvector(l_channel,ImL); gradvector(l_channel,ImL);
gradvector(a_channel,Ima); gradvector(a_channel,Ima);
gradvector(b_channel,Imb); gradvector(b_channel,Imb);
double res =0.0; double res =0.0;
for(unsigned int i=0;i<ImL.size();i++) for(unsigned int i=0;i<ImL.size();i++)
{ {
res=sqrt(pow(ImL[i],2) + pow(Ima[i],2) + pow(Imb[i],2))/100; res=sqrt(pow(ImL[i],2) + pow(Ima[i],2) + pow(Imb[i],2))/100;
Cg.push_back(res); Cg.push_back(res);
} }
lab.release(); lab.release();
l_channel.release(); l_channel.release();
a_channel.release(); a_channel.release();
b_channel.release(); b_channel.release();
ImL.clear(); ImL.clear();
Ima.clear(); Ima.clear();
Imb.clear(); Imb.clear();
} }
void Decolor::add_vector(vector < vector <int> > &comb, int r,int g,int b) void Decolor::add_vector(vector < vector <int> > &comb, int r,int g,int b)
{ {
static int idx =0; static int idx =0;
comb.push_back( vector <int>() ); comb.push_back( vector <int>() );
comb.at(idx).push_back( r ); comb.at(idx).push_back( r );
comb.at(idx).push_back( g ); comb.at(idx).push_back( g );
comb.at(idx).push_back( b ); comb.at(idx).push_back( b );
idx++; idx++;
} }
void Decolor::add_to_vector_poly(vector < vector <double> > &polyGrad, vector <double> &curGrad) void Decolor::add_to_vector_poly(vector < vector <double> > &polyGrad, vector <double> &curGrad)
{ {
static int idx1 =0; static int idx1 =0;
polyGrad.push_back( vector <double>() ); polyGrad.push_back( vector <double>() );
for(unsigned int i=0;i<curGrad.size();i++) for(unsigned int i=0;i<curGrad.size();i++)
polyGrad.at(idx1).push_back(curGrad[i]); polyGrad.at(idx1).push_back(curGrad[i]);
idx1++; idx1++;
} }
void Decolor::weak_order(Mat img, vector <double> &alf) void Decolor::weak_order(Mat img, vector <double> &alf)
{ {
Mat curIm = Mat(img.size(),CV_32FC1); int h = img.size().height;
Mat red = Mat(img.size(),CV_32FC1); int w = img.size().width;
Mat green = Mat(img.size(),CV_32FC1); double sizefactor;
Mat blue = Mat(img.size(),CV_32FC1); if((h + w) > 800)
{
for(int i=0;i<img.size().height;i++) sizefactor = (double)800/(h+w);
for(int j=0;j<img.size().width;j++) resize(img,img,Size(rounding(h*sizefactor),rounding(w*sizefactor)));
{ }
red.at<float>(i,j) = img.at<float>(i,j*3+2);
green.at<float>(i,j) = img.at<float>(i,j*3+1); Mat curIm = Mat(img.size(),CV_32FC1);
blue.at<float>(i,j) = img.at<float>(i,j*3+0); Mat red = Mat(img.size(),CV_32FC1);
} Mat green = Mat(img.size(),CV_32FC1);
Mat blue = Mat(img.size(),CV_32FC1);
vector <double> Rg;
vector <double> Gg; for(int i=0;i<img.size().height;i++)
vector <double> Bg; for(int j=0;j<img.size().width;j++)
{
vector <double> t1; red.at<float>(i,j) = img.at<float>(i,j*3+2);
vector <double> t2; green.at<float>(i,j) = img.at<float>(i,j*3+1);
vector <double> t3; blue.at<float>(i,j) = img.at<float>(i,j*3+0);
}
vector <double> tmp1;
vector <double> tmp2; vector <double> Rg;
vector <double> tmp3; vector <double> Gg;
vector <double> Bg;
gradvector(red,Rg);
gradvector(green,Gg); vector <double> t1;
gradvector(blue,Bg); vector <double> t2;
double level = .05; vector <double> t3;
for(unsigned int i=0;i<Rg.size();i++) vector <double> tmp1;
{ vector <double> tmp2;
if(Rg[i] > level) vector <double> tmp3;
t1.push_back(1.0);
else gradvector(red,Rg);
t1.push_back(0.0); gradvector(green,Gg);
} gradvector(blue,Bg);
for(unsigned int i=0;i<Gg.size();i++)
{ double level = .05;
if(Gg[i] > level)
t2.push_back(1.0); for(unsigned int i=0;i<Rg.size();i++)
else {
t2.push_back(0.0); if(Rg[i] > level)
} t1.push_back(1.0);
for(unsigned int i=0;i<Bg.size();i++) else
{ t1.push_back(0.0);
if(Bg[i] > level) }
t3.push_back(1.0); for(unsigned int i=0;i<Gg.size();i++)
else {
t3.push_back(0.0); if(Gg[i] > level)
} t2.push_back(1.0);
for(unsigned int i=0;i<Rg.size();i++) else
{ t2.push_back(0.0);
if(Rg[i] < -1.0*level) }
tmp1.push_back(1.0); for(unsigned int i=0;i<Bg.size();i++)
else {
tmp1.push_back(0.0); if(Bg[i] > level)
} t3.push_back(1.0);
for(unsigned int i=0;i<Gg.size();i++) else
{ t3.push_back(0.0);
if(Gg[i] < -1.0*level) }
tmp2.push_back(1.0); for(unsigned int i=0;i<Rg.size();i++)
else {
tmp2.push_back(0.0); if(Rg[i] < -1.0*level)
} tmp1.push_back(1.0);
for(unsigned int i=0;i<Bg.size();i++) else
{ tmp1.push_back(0.0);
if(Bg[i] < -1.0*level) }
tmp3.push_back(1.0); for(unsigned int i=0;i<Gg.size();i++)
else {
tmp3.push_back(0.0); if(Gg[i] < -1.0*level)
} tmp2.push_back(1.0);
for(unsigned int i =0 ;i < Rg.size();i++) else
alf.push_back(t1[i] * t2[i] * t3[i]); tmp2.push_back(0.0);
}
for(unsigned int i =0 ;i < Rg.size();i++) for(unsigned int i=0;i<Bg.size();i++)
alf[i] -= tmp1[i] * tmp2[i] * tmp3[i]; {
if(Bg[i] < -1.0*level)
double sum =0.0; tmp3.push_back(1.0);
for(unsigned int i=0;i<alf.size();i++) else
sum += abs(alf[i]); tmp3.push_back(0.0);
}
sum = (double)100*sum/alf.size(); for(unsigned int i =0 ;i < Rg.size();i++)
alf.push_back(t1[i] * t2[i] * t3[i]);
red.release();
green.release(); for(unsigned int i =0 ;i < Rg.size();i++)
blue.release(); alf[i] -= tmp1[i] * tmp2[i] * tmp3[i];
curIm.release();
double sum =0.0;
Rg.clear(); for(unsigned int i=0;i<alf.size();i++)
Gg.clear(); sum += abs(alf[i]);
Bg.clear();
sum = (double)100*sum/alf.size();
t1.clear();
t2.clear(); red.release();
t3.clear(); green.release();
blue.release();
tmp1.clear(); curIm.release();
tmp2.clear();
tmp3.clear(); Rg.clear();
Gg.clear();
Bg.clear();
t1.clear();
t2.clear();
t3.clear();
tmp1.clear();
tmp2.clear();
tmp3.clear();
} }
void Decolor::grad_system(Mat img, vector < vector < double > > &polyGrad, vector < double > &Cg, vector < vector <int> >& comb) void Decolor::grad_system(Mat img, vector < vector < double > > &polyGrad,
vector < double > &Cg, vector < vector <int> >& comb)
{ {
int h = img.size().height; int h = img.size().height;
int w = img.size().width; int w = img.size().width;
colorGrad(img,Cg);
Mat curIm = Mat(img.size(),CV_32FC1); double sizefactor;
Mat red = Mat(img.size(),CV_32FC1); if((h + w) > 800)
Mat green = Mat(img.size(),CV_32FC1); {
Mat blue = Mat(img.size(),CV_32FC1); sizefactor = (double)800/(h+w);
resize(img,img,Size(rounding(h*sizefactor),rounding(w*sizefactor)));
for(int i=0;i<img.size().height;i++) }
for(int j=0;j<img.size().width;j++)
{ h = img.size().height;
red.at<float>(i,j) = img.at<float>(i,j*3+2); w = img.size().width;
green.at<float>(i,j) = img.at<float>(i,j*3+1); colorGrad(img,Cg);
blue.at<float>(i,j) = img.at<float>(i,j*3+0);
} Mat curIm = Mat(img.size(),CV_32FC1);
Mat red = Mat(img.size(),CV_32FC1);
for(int r=0 ;r <=order; r++) Mat green = Mat(img.size(),CV_32FC1);
for(int g=0; g<=order;g++) Mat blue = Mat(img.size(),CV_32FC1);
for(int b =0; b <=order;b++)
{ for(int i=0;i<img.size().height;i++)
if((r+g+b)<=order && (r+g+b) > 0) for(int j=0;j<img.size().width;j++)
{ {
add_vector(comb,r,g,b); red.at<float>(i,j) = img.at<float>(i,j*3+2);
for(int i = 0;i<h;i++) green.at<float>(i,j) = img.at<float>(i,j*3+1);
for(int j=0;j<w;j++) blue.at<float>(i,j) = img.at<float>(i,j*3+0);
curIm.at<float>(i,j)= }
pow(red.at<float>(i,j),r)*pow(green.at<float>(i,j),g)*
pow(blue.at<float>(i,j),b); for(int r=0 ;r <=order; r++)
vector <double> curGrad; for(int g=0; g<=order;g++)
gradvector(curIm,curGrad); for(int b =0; b <=order;b++)
add_to_vector_poly(polyGrad,curGrad); {
} if((r+g+b)<=order && (r+g+b) > 0)
} {
add_vector(comb,r,g,b);
red.release(); for(int i = 0;i<h;i++)
green.release(); for(int j=0;j<w;j++)
blue.release(); curIm.at<float>(i,j)=
curIm.release(); pow(red.at<float>(i,j),r)*pow(green.at<float>(i,j),g)*
pow(blue.at<float>(i,j),b);
vector <double> curGrad;
gradvector(curIm,curGrad);
add_to_vector_poly(polyGrad,curGrad);
}
}
red.release();
green.release();
blue.release();
curIm.release();
} }
void Decolor::wei_update_matrix(vector < vector <double> > &poly, vector <double> &Cg, Mat &X) void Decolor::wei_update_matrix(vector < vector <double> > &poly, vector <double> &Cg, Mat &X)
{ {
Mat P = Mat(poly.size(),poly[0].size(), CV_32FC1); Mat P = Mat(poly.size(),poly[0].size(), CV_32FC1);
Mat A = Mat(poly.size(),poly.size(), CV_32FC1); Mat A = Mat(poly.size(),poly.size(), CV_32FC1);
for(unsigned int i =0;i<poly.size();i++) for(unsigned int i =0;i<poly.size();i++)
for(unsigned int j=0;j<poly[0].size();j++) for(unsigned int j=0;j<poly[0].size();j++)
P.at<float>(i,j) = poly[i][j]; P.at<float>(i,j) = poly[i][j];
Mat P_trans = P.t(); Mat P_trans = P.t();
Mat B = Mat(poly.size(),poly[0].size(), CV_32FC1); Mat B = Mat(poly.size(),poly[0].size(), CV_32FC1);
for(unsigned int i =0;i < poly.size();i++) for(unsigned int i =0;i < poly.size();i++)
{ {
for(unsigned int j=0;j<Cg.size();j++) for(unsigned int j=0;j<Cg.size();j++)
B.at<float>(i,j) = poly[i][j]*Cg[j]; B.at<float>(i,j) = poly[i][j]*Cg[j];
} }
A = P*P_trans; A = P*P_trans;
solve(A, B, X, DECOMP_NORMAL); solve(A, B, X, DECOMP_NORMAL);
P.release(); P.release();
A.release(); A.release();
B.release(); B.release();
} }
void Decolor::wei_inti(vector < vector <int> > &comb, vector <double> &wei) void Decolor::wei_inti(vector < vector <int> > &comb, vector <double> &wei)
{ {
vector <double> initRGB; vector <double> initRGB;
initRGB.push_back( .33 ); initRGB.push_back( .33 );
initRGB.push_back( .33 ); initRGB.push_back( .33 );
initRGB.push_back( .33 ); initRGB.push_back( .33 );
wei = product(comb,initRGB); wei = product(comb,initRGB);
vector <int> sum; vector <int> sum;
for(unsigned int i=0;i<comb.size();i++) for(unsigned int i=0;i<comb.size();i++)
sum.push_back(comb[i][0] + comb[i][1] + comb[i][2]); sum.push_back(comb[i][0] + comb[i][1] + comb[i][2]);
for(unsigned int i=0;i<sum.size();i++) for(unsigned int i=0;i<sum.size();i++)
{ {
if(sum[i] == 1) if(sum[i] == 1)
wei[i] = wei[i] * double(1); wei[i] = wei[i] * double(1);
else else
wei[i] = wei[i] * double(0); wei[i] = wei[i] * double(0);
} }
initRGB.clear(); initRGB.clear();
sum.clear(); sum.clear();
} }
void Decolor::grayImContruct(vector <double> &wei, Mat img, Mat &Gray) void Decolor::grayImContruct(vector <double> &wei, Mat img, Mat &Gray)
{ {
int h=img.size().height; int h=img.size().height;
int w=img.size().width; int w=img.size().width;
Mat red = Mat(img.size(),CV_32FC1); Mat red = Mat(img.size(),CV_32FC1);
Mat green = Mat(img.size(),CV_32FC1); Mat green = Mat(img.size(),CV_32FC1);
Mat blue = Mat(img.size(),CV_32FC1); Mat blue = Mat(img.size(),CV_32FC1);
for(int i=0;i<img.size().height;i++) for(int i=0;i<img.size().height;i++)
for(int j=0;j<img.size().width;j++) for(int j=0;j<img.size().width;j++)
{ {
red.at<float>(i,j) = img.at<float>(i,j*3+2); red.at<float>(i,j) = img.at<float>(i,j*3+2);
green.at<float>(i,j) = img.at<float>(i,j*3+1); green.at<float>(i,j) = img.at<float>(i,j*3+1);
blue.at<float>(i,j) = img.at<float>(i,j*3+0); blue.at<float>(i,j) = img.at<float>(i,j*3+0);
} }
int kk =0; int kk =0;
for(int r =0;r<=order;r++) for(int r =0;r<=order;r++)
for(int g=0;g<=order;g++) for(int g=0;g<=order;g++)
for(int b=0;b<=order;b++) for(int b=0;b<=order;b++)
if((r + g + b) <=order && (r+g+b) > 0) if((r + g + b) <=order && (r+g+b) > 0)
{ {
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++)
Gray.at<float>(i,j)=Gray.at<float>(i,j) + Gray.at<float>(i,j)=Gray.at<float>(i,j) +
wei[kk]*pow(red.at<float>(i,j),r)*pow(green.at<float>(i,j),g)* wei[kk]*pow(red.at<float>(i,j),r)*pow(green.at<float>(i,j),g)*
pow(blue.at<float>(i,j),b); pow(blue.at<float>(i,j),b);
kk=kk+1; kk=kk+1;
} }
double minval = INT_MAX; double minval = INT_MAX;
double maxval = INT_MIN; double maxval = INT_MIN;
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++)
{ {
if(Gray.at<float>(i,j) < minval) if(Gray.at<float>(i,j) < minval)
minval = Gray.at<float>(i,j); minval = Gray.at<float>(i,j);
if(Gray.at<float>(i,j) > maxval) if(Gray.at<float>(i,j) > maxval)
maxval = Gray.at<float>(i,j); maxval = Gray.at<float>(i,j);
} }
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++)
Gray.at<float>(i,j) = (Gray.at<float>(i,j) - minval)/(maxval - minval); Gray.at<float>(i,j) = (Gray.at<float>(i,j) - minval)/(maxval - minval);
red.release(); red.release();
green.release(); green.release();
blue.release(); blue.release();
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment