【opencv】【图像压缩】opencv下的DCT变换压缩图像

#include "stdafx.h"
#include<iostream>  
#include<sstream>
//#include <opencv2/core/core.hpp>  
//#include <opencv2/highgui/highgui.hpp>  
//#include<math.h>
//#include <opencv.h>
#include <opencv2/opencv.hpp>
#include <cv.h>
#include <string.h>
double T = 10;


using namespace cv;
using namespace std;

int main()
{
	// 读入一张图片（游戏原画）  
	Mat src = imread("sw1.png");
	resize(src,src,Size(src.cols/2*2,src.rows/2*2));
	imshow("Origin Image", src);
	int h = src.rows;
	int w = src.cols;

	//Mat yuvimg = src.clone();
	//从BGR空间转换到YUV空间（也可以不转换，直接在RGB空间）      
	Mat yuvimg(src.size(), CV_8UC3);
	cvtColor(src, yuvimg, CV_BGR2YUV); //定义YUV空间图像为yuvimage    
	Mat dst(src.size(), CV_64FC3);     //定义输出图像为dst    

									   //分割YUV通道    
	vector<Mat> channels;
	split(yuvimg, channels);

	//提取YUV颜色空间各通道      
	Mat Y = channels.at(0); imshow("Y image", Y);
	Mat U = channels.at(1); imshow("U image", U);
	Mat V = channels.at(2); imshow("V image", V);
	//waitKey(0);
	//DCT系数的三个通道    
	//Mat DCTY(src.size(), CV_64FC1);
	//Mat DCTU(src.size(), CV_64FC1);
	//Mat DCTV(src.size(), CV_64FC1);
	Mat DCTY(yuvimg.size(), CV_64FC1);
	Mat DCTU(yuvimg.size(), CV_64FC1);
	Mat DCTV(yuvimg.size(), CV_64FC1);
	//DCT变换    
	dct(Mat_<double>(Y), DCTY,0);
	dct(Mat_<double>(U), DCTU,0);
	dct(Mat_<double>(V), DCTV,0);

	//Y通道压缩      
	for (int i = 0; i < h; i++)
	{
		double *p = DCTY.ptr<double>(i);
		for (int j = 0; j < w; j++)
		{
			if (abs(p[j]) < T)
				p[j] = 0;
		}
	}

	//U通道压缩      
	for (int i = 0; i < h; i++)
	{
		double *p = DCTU.ptr<double>(i);
		for (int j = 0; j < w; j++)
		{
			if (abs(p[j]) < T)
				p[j] = 0;
		}
	}

	//V通道压缩      
	for (int i = 0; i < h; i++)
	{
		double *p = DCTV.ptr<double>(i);
		for (int j = 0; j < w; j++)
		{
			if (abs(p[j]) < T)
				p[j] = 0;
		}
	}
	Mat dstY(src.size(), CV_64FC1);
	Mat dstU(src.size(), CV_64FC1);
	Mat dstV(src.size(), CV_64FC1);

	//DCT逆变换    
	idct(DCTY, dstY);
	idct(DCTU, dstU);
	idct(DCTV, dstV);

	//merge方式1    
	//Mat planes[] = { Mat_<uchar>(dstB), Mat_<uchar>(dstG), Mat_<uchar>(dstR) };    
	//merge(planes, 3, yuvimg);     
	//cvtColor(yuvimg, dst, CV_YUV2BGR);    

	//merge方式2    
	channels.at(0) = Mat_<uchar>(dstY);
	channels.at(1) = Mat_<uchar>(dstU);
	channels.at(2) = Mat_<uchar>(dstV);
	merge(channels, yuvimg);

	//将压缩后图像从YUV空间重新转换到BGR空间    
	cvtColor(yuvimg, dst, CV_YUV2BGR);

	imshow("Recoverd Y image", Mat_<uchar>(dstY));
	imshow("Recoverd U image", Mat_<uchar>(dstU));
	imshow("Recoverd V image", Mat_<uchar>(dstV));
	imshow("DstImage", dst);
	stringstream s;
	s<<T;
	string t=s.str();
	imwrite(string("DstImage")+t+".png",dst);
	waitKey(0);
	return 0;
}