读取RGB及YUV格式图片并计算各自熵

读取RGB及YUV格式图片并计算各自熵

1.前言

工具：c语言
方法：利用程序编写实现读入一个RGB文件和一个YUV文件（示例文件down.yuv和down.rgb，分辨率均为256*256），计算R\G\B以及Y\U\V各三个通道的概率分布和各自的熵，、最后进行统计绘图。

2.实验思路

1.分别读取R、G、B（或Y、U、V）到数组中。
RGB文件 按每个像素BGR分量依次存放。
YUV文件 按照全部像素的Y数据块、U数据块和V数据块依次存放。
2. 分别统计R、G、B（或Y、U、V）三通道的颜色强度级的频数。
3. 别计算R、G、B（或Y、U、V）三通道的颜色强度级的概率，将概率写入文件。
4. 计算熵并输出。

3.实验内容

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 (int i =