【数据压缩】读入RGB文件并输出三个分量的熵

最新推荐文章于 2024-10-09 10:07:13 发布

Ddaeng.

最新推荐文章于 2024-10-09 10:07:13 发布

阅读量351

点赞数

分类专栏：数据压缩文章标签： RGB文件熵计算颜色分量概率信息理论

本文链接：https://blog.csdn.net/Uuuyuleo/article/details/123474753

版权

数据压缩专栏收录该内容

1 篇文章 0 订阅

订阅专栏

读入RGB文件并输出三个分量的熵

实现思路：

打开RGB文件，读出全部数据，计算RGB文件的大小。
分别设置三个分量的计数数组。
遍历一遍数据，将三个分量各颜色数值出现的次数存入数组中。
设置新数组，存入经计算得到的各颜色数值出现的概率。
计算三个颜色分量的熵。

代码实现：

#include <iostream>
#include <cmath>
#include <fstream>
using namespace std;

int main() {
    FILE *fp=fopen("test.rgb","rb");//打开rgb文件
    
    //计算文件大小 应为256*256*3
    if(!fp) return -1;
    fseek(fp,0,SEEK_END);
    long size=ftell(fp);
    cout<<"the size of test.rgb:"<<size<<endl;
    fseek(fp,0,SEEK_SET);
    
    //rgb数据存储到动态地址
    unsigned char *buffer=(unsigned char*)malloc(sizeof(unsigned char)*256*256*3);
    fread(buffer,sizeof(unsigned char),256*256*3,fp);
    
    //设置计数数组
    int B[256]={0};
    int G[256]={0};
    int R[256]={0};

    //计数
    int b=0,g=0,r=0;
    for(int i=0;i<256*256;i++){
        b=*(buffer+3*i);
        B[b]++;
        g=*(buffer+3*i+1);
        G[g]++;
        r=*(buffer+3*i+2);
        R[r]++;
    }

    //计算概率
    double pB[256]={0};
    double pG[256]={0};
    double pR[256]={0};
    for(int i=0;i<256;i++){
        if(B[i]!=0) pB[i]=double(B[i])/(256*256);
        else pB[i]=0;
        if(G[i]!=0) pG[i]=double(G[i])/(256*256);
        else pG[i]=0;
        if(R[i]!=0) pR[i]=double(R[i])/(256*256);
        else pR[i]=0;
    }

    //计算熵
    double hB=0,hG=0,hR=0;
    for(int i=0;i<256;i++){
        if(pB[i]!=0) hB+=pB[i]*(log(1/pB[i])/log(2));
        if(pG[i]!=0) hG+=pG[i]*(log(1/pG[i])/log(2));
        if(pR[i]!=0) hR+=pR[i]*(log(1/pR[i])/log(2));
    }

    //输出熵
    cout<<"Red Source Entropy is "<<hR<<endl;
    cout<<"Green Source Entropy is "<<hG<<endl;
    cout<<"Blue Source Entropy is "<<hB<<endl;

    fclose(fp);
    free(buffer);

    return 0;
}