利用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 (