计算YUV和RGB图像各通道的概率分布和熵

利用C语言计算YUV和RGB各通道概率分布并求熵

一、实验目的

利用程序编写实现读入一个RGB文件和一个YUV文件（示例文件down.yuv和down.rgb，分辨率均为256*256），计算R\G\B以及Y\U\V各三个通道的概率分布和各自的熵，结果输出到csv文件中，最后导入Excel进行统计绘图。

以下为示例图像：

• 注：rgb文件按每个像素BGR分量依次存放；YUV格式按照全部像素的Y数据块、U数据块和V数据块依次存放。

二、实验内容

1.RGB图像分析

程序：

#include<stdio.h>
#include<iostream>
#include<math.h>

using namespace std;
#define A  65536 //分辨率为256*256
int main()
{
   
	//打开文件,创建输出文件
	FILE* image, * red, * green, * blue;
	fopen_s(&image, "D:\\学校文档\\2021春\\数据压缩\\down.rgb", "rb");
	fopen_s(&red, "D:\\学校文档\\2021春\\数据压缩\\R.txt", "w");
	fopen_s(&green, "D:\\学校文档\\2021春\\数据压缩\\G.txt", "w");
	fopen_s(&blue, "D:\\学校文档\\2021春\\数据压缩\\B.txt", "w");
	
	unsigned char R[A] = {
    0 }, G[A] = {
    0 }, B[A] = {
    0 };//定义R、G、B分量
	
	double R_F[256] = {
    0 }, G_F[256] = {
    0 }, B_F[256] = {
    0 };//定义频率分量
	
	double R_S = 0, G_S = 0, B_S = 0;//定义熵
	
	unsigned char sum[256 * 256 * 3];//分别读取R、G、B三个分量到数组中
	fread(sum, 1, 256 * 256 * 3, image);
	for (int i = 0, j = 0; i < 256 * 256 * 3; i = i + 3, j++)
	{
   
		B[j] = *(sum + i);
		G[j] = *(sum + i + 1);
		R[j] = *(sum + i + 2);
	}
	//计数三通道各颜色值次数
	for (