首页 > 代码库 > OpenCV——PS 图层混合算法 (四)
OpenCV——PS 图层混合算法 (四)
具体的算法原理可以参考
PS图层混合算法之四(亮光, 点光, 线性光, 实色混合)
// PS_Algorithm.h
#ifndef PS_ALGORITHM_H_INCLUDED
#define PS_ALGORITHM_H_INCLUDED
#include <iostream>#include <string>
#include "cv.h"
#include "highgui.h"
#include "cxmat.hpp"
#include "cxcore.hpp"
using namespace std;using namespace cv;
#endif // PS_ALGORITHM_H_INCLUDED
// main function
#include "PS_Algorithm.h"
void Vivid_Lighten(Mat& src1, Mat& src2, Mat& dst);
void Pin_Lighten(Mat& src1, Mat& src2, Mat& dst);
void Linear_Lighten(Mat& src1, Mat& src2, Mat& dst);
void Hard_mix(Mat& src1, Mat& src2, Mat& dst);
int main(void)
{
Mat Origin_Image1;
Mat Origin_Image2;
Origin_Image1=imread("2.jpg");
Origin_Image2=imread("3.jpg");
Mat Image_up(Origin_Image1.size(),CV_32FC3);
Mat Image_down(Origin_Image2.size(), CV_32FC3);
Origin_Image1.convertTo(Image_up,CV_32FC3);
Origin_Image2.convertTo(Image_down,CV_32FC3);
Image_up=Image_up/255;
Image_down=Image_down/255;
Mat Image_mix(Image_up);
//Vivid_Lighten(Image_up, Image_down, Image_mix);
//Pin_Lighten(Image_up, Image_down, Image_mix);
//Linear_Lighten(Image_up, Image_down, Image_mix);
//Hard_mix(Image_up, Image_down, Image_mix);
namedWindow("Img", CV_WINDOW_AUTOSIZE);
imshow("Img",Image_mix);
waitKey();
cvDestroyWindow("Img");
cout<<"All is well."<<endl;
return 0;
}
//Vivid Lighten
void Vivid_Lighten(Mat& src1, Mat& src2, Mat& dst)
{
float a=0;
float b=0;
for(int index_row=0; index_row<src1.rows; index_row++)
{
for(int index_col=0; index_col<src1.cols; index_col++)
{
for(int index_c=0; index_c<3; index_c++)
{
a=src1.at<Vec3f>(index_row, index_col)[index_c];
b=src2.at<Vec3f>(index_row, index_col)[index_c];
if(a<=0.5)
{
dst.at<Vec3f>(index_row, index_col)[index_c]=1-(1-b)/(2*a);
}
else
{
dst.at<Vec3f>(index_row, index_col)[index_c]=b/(2*(1-a));
}
}
}
}
}
// Pin lighten
void Pin_Lighten(Mat& src1, Mat& src2, Mat& dst)
{
float a=0;
float b=0;
for(int index_row=0; index_row<src1.rows; index_row++)
{
for(int index_col=0; index_col<src1.cols; index_col++)
{
for(int index_c=0; index_c<3; index_c++)
{
a=src1.at<Vec3f>(index_row, index_col)[index_c];
b=src2.at<Vec3f>(index_row, index_col)[index_c];
if(b<=2*a-1)
{
dst.at<Vec3f>(index_row, index_col)[index_c]=2*a-1;
}
else if(b<=2*a)
{
dst.at<Vec3f>(index_row, index_col)[index_c]=b;
}
else
{
dst.at<Vec3f>(index_row, index_col)[index_c]=2*a;
}
}
}
}
}
// Linear Lighten
void Linear_Lighten(Mat& src1, Mat& src2, Mat& dst)
{
dst=src2+2*src1-1;
}
// Hard mix
void Hard_mix(Mat& src1, Mat& src2, Mat& dst)
{
float a=0;
float b=0;
for(int index_row=0; index_row<src1.rows; index_row++)
{
for(int index_col=0; index_col<src1.cols; index_col++)
{
for(int index_c=0; index_c<3; index_c++)
{
a=src1.at<Vec3f>(index_row, index_col)[index_c];
b=src2.at<Vec3f>(index_row, index_col)[index_c];
if(a<1-b)
{
dst.at<Vec3f>(index_row, index_col)[index_c]=0.0;
}
else
{
dst.at<Vec3f>(index_row, index_col)[index_c]=1.0;
}
}
}
}
}