OpenCV读取图片，DCT变换

最新推荐文章于 2024-12-10 00:26:29 发布

东山阿强

最新推荐文章于 2024-12-10 00:26:29 发布

阅读量1.1k

点赞数 4

CC 4.0 BY-SA版权

分类专栏：编解码文章标签： opencv cv

本文链接：https://blog.csdn.net/weixin_43306331/article/details/111051973

编解码专栏收录该内容

2 篇文章

订阅专栏

#include <iostream>  
#include <string.h>
#include <opencv2/core/core.hpp>  
#include <opencv2/highgui/highgui.hpp>  
#include <opencv2/core/mat.hpp>
#include <opencv2/imgproc/types_c.h>
#include <opencv2/imgproc/imgproc.hpp>
#include<opencv2/opencv.hpp>//如果要包含所有库
#define PI 3.1415926

using namespace cv;
using namespace std;

void initDctMat(Mat& A) {
	for (int i = 0; i < 8; ++i)
		for (int j = 0; j < 8; ++j)
		{
			float a;
			if (i == 0)
				a = sqrt(1.0 / 8.0);
			else
				a = sqrt(2.0 / 8.0);
			A.ptr<float>(i)[j] = a * cos((j + 0.5) * PI * i / 8);
		}
}

Mat dct_block(const Mat& src, const Mat& A, const Mat& mask) {
	int rows = src.rows;
	int cols = src.cols;

	Mat res = src.clone();
	for(int i = 0; i < rows; i+= 8)
		for (int j = 0; j < cols; j += 8) {
			res(Range(i, i + 8), Range(j, j + 8)) = A *res(Range(i, i + 8), Range(j, j + 8)) * A.t(); //dct
			res(Range(i, i + 8), Range(j, j + 8)) /= mask; //quantative
		}
	return res;
}

Mat idct_block(const Mat& src, const Mat& A, const Mat& mask) {
	int rows = src.rows;
	int cols = src.cols;

	Mat res = src.clone();
	for (int i = 0; i < rows; i += 8)
		for (int j = 0; j < cols; j += 8) {
			res(Range(i, i + 8), Range(j, j + 8)) = res(Range(i, i + 8), Range(j, j + 8)).mul(mask); //X = ATXA, idct
		    res(Range(i, i + 8), Range(j, j + 8)) = A.t() * res(Range(i, i + 8), Range(j, j + 8)) * A;
		}
	return res;
}

void main()
{

	Mat A(Size(8, 8), CV_32F);
	float msk[8][8] = { {16,11,10,16,24,40,51,61},{12,12,14,19,26,58,60,55},{14,13,16,24,40,57,69,56},{14,17,22,29,51,87,80,62},{18,22,37,56,68,109,103,77},{24,35,55,64,81,104,113,92},{49,64,78,87,103,121,120,101},{72,92,95,98,112,100,103,99} };
	Mat mask(8, 8, CV_32F, msk);

	initDctMat(A);

	Mat src = imread("C:/Users/37075/source/repos/encoder/2.jpg");
	Mat YUVsrc;
	Mat res;
	cvtColor(src, YUVsrc, CV_BGR2YUV);
	vector<Mat> yuv;
	split(YUVsrc, yuv);
	Mat y, u, v;
	y = yuv[0];
	u = yuv[1];
	v = yuv[2];
	y.convertTo(y, CV_32F, 1.0 / 255);
    u.convertTo(u, CV_32F, 1.0 / 255);
	v.convertTo(v, CV_32F, 1.0 / 255);
	
	Mat ydct, udct, vdct, yidct, uidct, vidct;
	ydct = dct_block(y, A, mask);
	udct = dct_block(u, A, mask);
	vdct = dct_block(v, A, mask);
	yidct = idct_block(ydct, A, mask);
	uidct = idct_block(udct, A, mask);
	vidct = idct_block(vdct, A, mask);
	//cout << ydct << endl;
	//dct(y, ydct);
	//dct(u, udct);
	//dct(v, vdct);
	//idct(ydct, yidct);
	//idct(udct, uidct);
	//idct(vdct, vidct);
	vector<Mat> idct = { yidct, uidct, vidct };
	Mat tmp;
	merge(idct, tmp);
	cvtColor(tmp, res, CV_YUV2BGR);

	vector<Mat> dct = { ydct, udct, vdct };
	merge(dct, tmp);
	Mat res2;
	cvtColor(tmp, res2, CV_YUV2BGR);


	imshow("res", res);
	imshow("resdct", res2);
	res.convertTo(res, CV_8UC1, 255);
	res2.convertTo(res2, CV_8UC1, 255);
	imwrite("res.jpg", res);
	imwrite("resdct.jpg", res2);
	waitKey();

}