int main(int argc,char* argv[])
{
char src[50];
checkfile(src);
IplImage *I = cvLoadImage(src);
float sigma = .02;
int maxIter = 10;
int iterCount = 0;
if(I->nChannels !=3)
{
printf("Input RGB Image\n");
exit(0);
}
Decolor obj;
obj.display("Original Image",I);
int h1 = I->height;
int w1 = I->width;
IplImage *dst = cvCreateImage(cvGetSize(I), IPL_DEPTH_32F, 3 );
IplImage *final1 = cvCreateImage(cvGetSize(I), 8, 3 );
double sizefactor;
IplImage *img;
if((h1+w1) > 900)
{
sizefactor = (double)900/(h1+w1);
IplImage *Im = cvCreateImage(cvSize(obj.rounding(w1*sizefactor) , obj.rounding(h1*sizefactor)),I->depth, I->nChannels);
cvResize(I, Im);
img = cvCreateImage(cvGetSize(Im), IPL_DEPTH_32F, 3 );
cvConvertScale(Im,img,1.0/255.0,0.0);
cvReleaseImage(&Im);
}
else
{
img = cvCreateImage(cvGetSize(I), IPL_DEPTH_32F, 3 );
cvConvertScale(I,img,1.0/255.0,0.0);
}
int h = img->height;
int w = img->width;
int channel = img->nChannels;
IplImage *orig_gray = cvCreateImage(cvGetSize(I), 8, 1 );
cvCvtColor(I,orig_gray,CV_BGR2GRAY);
obj.display("Gray Image",orig_gray);
cvSaveImage("orig_gray.jpg",orig_gray);
obj.init();
vector <double> Cg;
vector < vector <double> > polyGrad;
vector < vector <double> > bc;
vector < vector < int > > comb;
vector <double> alf;
obj.grad_system(img,polyGrad,Cg,comb);
obj.weak_order(img,alf);
Mat Mt = Mat(polyGrad.size(),polyGrad[0].size(), CV_32FC1);
obj.wei_update_matrix(polyGrad,Cg,Mt);
vector <double> wei;
obj.wei_inti(comb,wei);
//////////////////////////////// main loop starting ////////////////////////////////////////
while (iterCount < maxIter)
{
iterCount +=1;
vector <double> G_pos;
vector <double> G_neg;
vector <double> temp;
vector <double> temp1;
double val = 0.0;
for(int i=0;i< polyGrad[0].size();i++)
{
val = 0.0;
for(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]);
}
double ans = 0.0;
double ans1 = 0.0;
for(int i =0;i<alf.size();i++)
{
ans = ((1 + alf[i])/2) * exp((-1.0 * 0.5 * pow(temp[i],2))/pow(sigma,2));
ans1 =((1 - alf[i])/2) * exp((-1.0 * 0.5 * pow(temp1[i],2))/pow(sigma,2));
G_pos.push_back(ans);
G_neg.push_back(ans1);
}
vector <double> EXPsum;
vector <double> EXPterm;
for(int i = 0;i<G_pos.size();i++)
EXPsum.push_back(G_pos[i]+G_neg[i]);
vector <double> temp2;
for(int i=0;i<EXPsum.size();i++)
{
if(EXPsum[i] == 0)
temp2.push_back(1.0);
else
temp2.push_back(0.0);
}
for(int i =0; i < G_pos.size();i++)
EXPterm.push_back((G_pos[i] - G_neg[i])/(EXPsum[i] + temp2[i]));
double val1 = 0.0;
vector <double> wei1;
for(int i=0;i< polyGrad.size();i++)
{
val1 = 0.0;
for(int j =0;j<polyGrad[0].size();j++)
{
val1 = val1 + (Mt.at<float>(i,j) * EXPterm[j]);
}
wei1.push_back(val1);
}
for(int i =0;i<wei.size();i++)
wei[i] = wei1[i];
G_pos.clear();
G_neg.clear();
temp.clear();
temp1.clear();
EXPsum.clear();
EXPterm.clear();
temp2.clear();
wei1.clear();
}
IplImage *Gray = cvCreateImage(cvGetSize(img), IPL_DEPTH_32F, 1 );
obj.grayImContruct(wei, img, Gray);
IplImage* O = cvCreateImage(cvGetSize(img),8,1);
cvConvertScale(Gray,O,255);
cvSaveImage("Grayimage.jpg",O);
obj.display("GrayImage",O);
//////////////////////////////////////// Contrast Boosting /////////////////////////////////////////////////////
IplImage *lab = cvCreateImage(cvGetSize(img), 8, 3 );
IplImage *contrast = cvCreateImage(cvGetSize(img), 8, 3 );
IplImage *l_channel = cvCreateImage(cvGetSize(img), 8, 1 );
IplImage *a_channel = cvCreateImage(cvGetSize(img), 8, 1 );
IplImage *b_channel = cvCreateImage(cvGetSize(img), 8, 1 );
cvCvtColor(I,lab,CV_BGR2Lab);
cvCvtPixToPlane(lab, l_channel, a_channel, b_channel,0);
for(int i =0;i<h1;i++)
for(int j=0;j<w1;j++)
{
CV_IMAGE_ELEM(l_channel,uchar,i,j) = 255.0*CV_IMAGE_ELEM(Gray,float,i,j);
}
for(int i =0;i<h1;i++)
for(int j=0;j<w1;j++)
{
CV_IMAGE_ELEM(lab,uchar,i,j*3+0) = CV_IMAGE_ELEM(l_channel,uchar,i,j);
CV_IMAGE_ELEM(lab,uchar,i,j*3+1) = CV_IMAGE_ELEM(a_channel,uchar,i,j);
CV_IMAGE_ELEM(lab,uchar,i,j*3+2) = CV_IMAGE_ELEM(b_channel,uchar,i,j);
}
cvCvtColor(lab,contrast,CV_Lab2BGR);
obj.display("Color boost",contrast);
cvSaveImage("Colorboost.jpg",contrast);
cvReleaseImage(&I);
cvReleaseImage(&final1);
cvReleaseImage(&dst);
cvReleaseImage(&lab);
cvReleaseImage(&contrast);
cvReleaseImage(&l_channel);
cvReleaseImage(&a_channel);
cvReleaseImage(&b_channel);
cvReleaseImage(&O);
cvReleaseImage(&Gray);
Cg.clear();
polyGrad.clear();
bc.clear();
comb.clear();
alf.clear();
wei.clear();
cvWaitKey(0);
}
void cv::decolor(InputArray _src, OutputArray _dst, OutputArray _color_boost)
{
Mat I = _src.getMat();
_dst.create(I.size(), CV_8UC1);
Mat dst = _dst.getMat();
_color_boost.create(I.size(), CV_8UC3);
Mat color_boost = _color_boost.getMat();
if(!I.data )
{
cout << "Could not open or find the image" << endl ;
return;
}
if(I.channels() !=3)
{
cout << "Input Color Image" << endl;
return;
}
// Parameter Setting
int maxIter = 15;
int iterCount = 0;
double tol = .0001;
double E = 0;
double pre_E = std::numeric_limits<double>::infinity();
Decolor obj;
Mat img;
img = Mat(I.size(),CV_32FC3);
I.convertTo(img,CV_32FC3,1.0/255.0);
// Initialization
obj.init();
vector <double> Cg;
vector < vector <double> > polyGrad;
vector < vector < int > > comb;
vector <double> alf;
obj.grad_system(img,polyGrad,Cg,comb);
obj.weak_order(img,alf);
// Solver
Mat Mt = Mat((int)polyGrad.size(),(int)polyGrad[0].size(), CV_32FC1);
obj.wei_update_matrix(polyGrad,Cg,Mt);
vector <double> wei;
obj.wei_inti(comb,wei);
//////////////////////////////// main loop starting ////////////////////////////////////////
while(sqrt(pow(E-pre_E,2)) > tol)
{
iterCount +=1;
pre_E = E;
vector <double> G_pos;
vector <double> G_neg;
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]);
}
double pos = 0.0;
double neg = 0.0;
for(unsigned int i =0;i<alf.size();i++)
{
pos = ((1 + alf[i])/2) * exp((-1.0 * 0.5 * pow(temp[i],2))/pow(obj.sigma,2));
neg = ((1 - alf[i])/2) * exp((-1.0 * 0.5 * pow(temp1[i],2))/pow(obj.sigma,2));
G_pos.push_back(pos);
G_neg.push_back(neg);
}
vector <double> EXPsum;
vector <double> EXPterm;
for(unsigned int i = 0;i<G_pos.size();i++)
EXPsum.push_back(G_pos[i]+G_neg[i]);
vector <double> temp2;
for(unsigned int i=0;i<EXPsum.size();i++)
{
if(EXPsum[i] == 0)
temp2.push_back(1.0);
else
temp2.push_back(0.0);
}
for(unsigned int i =0; i < G_pos.size();i++)
EXPterm.push_back((G_pos[i] - G_neg[i])/(EXPsum[i] + temp2[i]));
double val1 = 0.0;
vector <double> wei1;
for(unsigned int i=0;i< polyGrad.size();i++)
{
val1 = 0.0;
for(unsigned int j =0;j<polyGrad[0].size();j++)
{
val1 = val1 + (Mt.at<float>(i,j) * EXPterm[j]);
}
wei1.push_back(val1);
}
for(unsigned int i =0;i<wei.size();i++)
wei[i] = wei1[i];
E = obj.energyCalcu(Cg,polyGrad,wei);
if(iterCount > maxIter)
break;
G_pos.clear();
G_neg.clear();
temp.clear();
temp1.clear();
EXPsum.clear();
EXPterm.clear();
temp2.clear();
wei1.clear();
}
Mat Gray = Mat::zeros(img.size(),CV_32FC1);
obj.grayImContruct(wei, img, Gray);
Gray.convertTo(dst,CV_8UC1,255);
/////////////////////////////////// Contrast Boosting /////////////////////////////////
Mat lab = Mat(img.size(),CV_8UC3);
Mat color = Mat(img.size(),CV_8UC3);
cvtColor(I,lab,COLOR_BGR2Lab);
vector <Mat> lab_channel;
split(lab,lab_channel);
dst.copyTo(lab_channel[0]);
merge(lab_channel,lab);
cvtColor(lab,color_boost,COLOR_Lab2BGR);
}
void cv::decolor(InputArray _src, OutputArray _dst, OutputArray _color_boost)
{
CV_INSTRUMENT_REGION();
Mat I = _src.getMat();
_dst.create(I.size(), CV_8UC1);
Mat dst = _dst.getMat();
_color_boost.create(I.size(), CV_8UC3);
Mat color_boost = _color_boost.getMat();
CV_Assert(!I.empty() && (I.channels()==3));
// Parameter Setting
const int maxIter = 15;
const double tol = .0001;
int iterCount = 0;
double E = 0;
double pre_E = std::numeric_limits<double>::infinity();
Mat img;
I.convertTo(img, CV_32FC3, 1.0/255.0);
// Initialization
Decolor obj;
vector <double> Cg;
vector < vector <double> > polyGrad;
vector <Vec3i> comb;
vector <double> alf;
obj.grad_system(img,polyGrad,Cg,comb);
obj.weak_order(img,alf);
// Solver
Mat Mt = Mat(int(polyGrad.size()),int(polyGrad[0].size()), CV_32FC1);
obj.wei_update_matrix(polyGrad,Cg,Mt);
vector <double> wei;
obj.wei_inti(comb,wei);
//////////////////////////////// main loop starting ////////////////////////////////////////
vector <double> G_pos(alf.size());
vector <double> G_neg(alf.size());
vector <double> EXPsum(G_pos.size());
vector <double> EXPterm(G_pos.size());
vector <double> temp(polyGrad[0].size());
vector <double> temp1(polyGrad[0].size());
vector <double> temp2(EXPsum.size());
vector <double> wei1(polyGrad.size());
while(sqrt(pow(E-pre_E,2)) > tol)
{
iterCount +=1;
pre_E = E;
for(size_t i=0; i<polyGrad[0].size(); i++)
{
double val = 0.0;
for(size_t j=0; j<polyGrad.size(); j++)
val = val + (polyGrad[j][i] * wei[j]);
temp[i] = val - Cg[i];
temp1[i] = val + Cg[i];
}
for(size_t i=0; i<alf.size(); i++)
{
const double sqSigma = obj.sigma * obj.sigma;
const double pos = ((1 + alf[i])/2) * exp(-1.0 * 0.5 * (temp[i] * temp[i]) / sqSigma);
const double neg = ((1 - alf[i])/2) * exp(-1.0 * 0.5 * (temp1[i] * temp1[i]) / sqSigma);
G_pos[i] = pos;
G_neg[i] = neg;
}
for(size_t i=0; i<G_pos.size(); i++)
EXPsum[i] = G_pos[i]+G_neg[i];
for(size_t i=0; i<EXPsum.size(); i++)
temp2[i] = (EXPsum[i] == 0) ? 1.0 : 0.0;
for(size_t i=0; i<G_pos.size(); i++)
EXPterm[i] = (G_pos[i] - G_neg[i])/(EXPsum[i] + temp2[i]);
for(int i=0; i<int(polyGrad.size()); i++)
{
double val1 = 0.0;
for(int j=0; j<int(polyGrad[0].size()); j++)
{
val1 = val1 + (Mt.at<float>(i,j) * EXPterm[j]);
}
wei1[i] = val1;
}
for(size_t i=0; i<wei.size(); i++)
wei[i] = wei1[i];
E = obj.energyCalcu(Cg, polyGrad, wei);
if(iterCount > maxIter)
break;
}
Mat Gray = Mat::zeros(img.size(),CV_32FC1);
obj.grayImContruct(wei, img, Gray);
Gray.convertTo(dst,CV_8UC1,255);
/////////////////////////////////// Contrast Boosting /////////////////////////////////
Mat lab;
cvtColor(I,lab,COLOR_BGR2Lab);
vector <Mat> lab_channel;
split(lab,lab_channel);
dst.copyTo(lab_channel[0]);
merge(lab_channel,lab);
cvtColor(lab,color_boost,COLOR_Lab2BGR);
}
