opencv 基于灰度直方图的颜色迁移

最新推荐文章于 2024-07-24 08:58:44 发布

lrl940607

最新推荐文章于 2024-07-24 08:58:44 发布

阅读量581

点赞数

分类专栏： opencv 文章标签： opencv 颜色迁移

本文链接：https://blog.csdn.net/lrl940607/article/details/50461113

版权

opencv 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

//opencv 基于灰度直方图的颜色迁移


#include <opencv2/opencv.hpp>
#include <opencv2/legacy/compat.hpp>
using namespace std;

void FillWhite(IplImage *pImage)
{
    cvRectangle(pImage, cvPoint(0, 0), cvPoint(pImage->width, pImage->height), CV_RGB(255, 255, 255), CV_FILLED);
}
// 创建灰度图像的直方图
CvHistogram* CreateGrayImageHist(IplImage **ppImage)
{
    int nHistSize = 256;
    float fRange[] = {0, 255};  //灰度级的范围
    float *pfRanges[] = {fRange};
    CvHistogram *pcvHistogram = cvCreateHist(1, &nHistSize, CV_HIST_ARRAY, pfRanges);
    cvCalcHist(ppImage, pcvHistogram);
    return pcvHistogram;
}
float a[256];
void TurnToPDF(CvHistogram *pcvHistogram, IplImage *pSrcImage){
    float s = 0;
    for (int i = 0; i < 256; i++) {
        s = s + cvQueryHistValue_1D(pcvHistogram, i);
        a[i] = s/pSrcImage->width/pSrcImage->height;
    }
}
float b[256];
void TurnToPDF2(CvHistogram *pcvHistogram, IplImage *pSrcImage){
    float s = 0;
    for (int i = 0; i < 256; i++) {
        s = s + cvQueryHistValue_1D(pcvHistogram, i);
        b[i] = s/pSrcImage->width/pSrcImage->height;
    }
}
int matchtable[256];
void FindMatch(){
    for (int i = 0; i< 256; i++) {
        for (int j = 0; j < 255; j++) {
            if (a[i]>=b[j]&&a[i]<=b[j+1]) {
                if (a[i]-b[j]<=b[j+1]-a[i]) {
                    matchtable[i] = j;
                }else matchtable[i] = j+1;
            }else if (a[i]>b[j+1]) matchtable[i] = 255;
        }
        cout<<"matchtable:i = "<<i<<"; j = "<<matchtable[i]<<endl;
    }
    
}
int main( int argc, char** argv )
{
    // 从文件中加载原图
    IplImage *pSrcImage = cvLoadImage("DIP-Lecture2-Point Processing_430309674.jpg", CV_LOAD_IMAGE_UNCHANGED);
    IplImage *pScrRGB[3];
    pScrRGB[0] = cvCreateImage(cvGetSize(pSrcImage), IPL_DEPTH_8U, 1);
    pScrRGB[1] = cvCreateImage(cvGetSize(pSrcImage), IPL_DEPTH_8U, 1);
    pScrRGB[2] = cvCreateImage(cvGetSize(pSrcImage), IPL_DEPTH_8U, 1);
    cvSplit(pSrcImage,pScrRGB[0],pScrRGB[1],pScrRGB[2],NULL);
    
    IplImage *pSrcImage2 = cvLoadImage("DIP-Lecture2-Point Processing_2.jpg", CV_LOAD_IMAGE_UNCHANGED);
    IplImage *pScrRGB2[3];
    pScrRGB2[0] = cvCreateImage(cvGetSize(pSrcImage2), IPL_DEPTH_8U, 1);
    pScrRGB2[1] = cvCreateImage(cvGetSize(pSrcImage2), IPL_DEPTH_8U, 1);
    pScrRGB2[2] = cvCreateImage(cvGetSize(pSrcImage2), IPL_DEPTH_8U, 1);
    cvSplit(pSrcImage2,pScrRGB2[0],pScrRGB2[1],pScrRGB2[2],NULL);

    IplImage *imgBuf = cvCreateImage(cvGetSize(pSrcImage), IPL_DEPTH_8U, 3);   //通道为3
    cvConvert(pSrcImage, imgBuf);
    for (int n = 0; n < 3; n++) {
        CvHistogram *pcvHistogram = CreateGrayImageHist(&pScrRGB[n]);
        TurnToPDF(pcvHistogram,pScrRGB[n]);
        CvHistogram *pcvHistogram2 = CreateGrayImageHist(&pScrRGB2[n]);
        TurnToPDF2(pcvHistogram2,pScrRGB2[n]);
        FindMatch();
        
        for (int i = 0; i < pSrcImage->height; i++)
        {
            for(int j = 0; j < pSrcImage->width; j++)
            {
                int r = ((uchar *)(pSrcImage->imageData +  i * pSrcImage->widthStep ))[j * pSrcImage->nChannels + 2-n];   //改变该像素R的颜色分量
                ((uchar *)(pSrcImage->imageData +  i * pSrcImage->widthStep ))[j * pSrcImage->nChannels + 2-n] = matchtable[r];
            }
        }
    }

    cvSaveImage("test1.jpg", pSrcImage);
  
    return 0;  
}