海思AI芯片(Hi3519A/3559A)方案学习（十一）分析bgr文件格式

3519A nnie SDK只支持BGR以及yuv格式的数据输入。那么当手头刚好只有jpeg/png等图片文件时，怎么来进行目标识别呢？一般的思路是用opencv来读取图片文件，然后再转换成BGR/YUV等格式，最后扔到NNIE模块来进行深度学习推理。
本系列为华为海思海思AI芯片(Hi3519A/3559A)方案学习系列之十一，系列文章请见底部。
作者：ltshan139

BGR文件数据格式

nnie sample里面提供的测试图片有后缀名为bgr的文件，例如dog_bick_car.bgr。在sample code里面直接通过fopen fread之类C文件API来读取二进制数据到buffer，然后再调用nnie forward API进行图像识别。

可是实际应用中，拿到的图片往往是jpeg/png等格式，所以需要先调用opencv库函数进行转换。而要正确的转换，就先要知道bgr文件的数据格式，亦即GBG的排列顺序。下面代码就是将sample里面的bgr按bgrbgrbgr... ...格式重构成Mat类型，然后进行显示，看看显示跟预期是否一致。

  FILE *fp;
    fp = fopen("D:/work/dog_bike_car_416x416.bgr", "rb");
    
        unsigned char img_data[416x416x3];
    fread(img_data, 1, 416x416x3, fp);
    fclose(fp);
 
        Mat img = Mat(416, 416, CV_8UC3, img_data);
        imshow("hello", img);

结果发现，显示不正确，如下所示：

后来，通过试验，发现它是按BBBB....GGGG....RRRR...排列的。其相应读取转换代码如下：

//img_data是按BBBB...GGGG...RRRR...排列的（bgr文件格式）
    //img_data_conv按BGRBGRBGR... ...排列的（OPencv Mat排列格式）
    unsigned char img_data[416 * 416 * 3], img_data_conv[416 * 416 * 3];
 
    FILE *fp;
    fp = fopen("D:/work/dog_bike_car_416x416.bgr", "rb");
    int channels = 3;
    int width = 416;
    int height = 416;
    
    fread(img_data, 1, width*height*channels, fp);
    fclose(fp);
    
    for (int h = 0; h < height; h++)
        for (int w = 0; w < width; w++)
            for (int c = 0; c < channels; c++)
            {
                img_data_conv[h * width * channels + w * channels + c] = img_data[(height * width) * c + h * width + w];
 
            }
        
    Mat img_test = Mat(height, width, CV_8UC3, img_data_conv);
        imshow("hello", img_test);

其显示如下，是正确的。

小结

上面代码写的比较粗糙，硬代码很多，也没考虑效率，大家理解就行。此外，本文主要是通过实验来明确海思sample里面bgr文件的数据格式，并没有提供bgrbgr...转换成b...g...r...代码，但是参考上面代码就应该很容易实现，说白了就是一逆过程。

版权声明：本文为CSDN博主「ltshan139」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/avideointerfaces/article/details/89367139

海思AI芯片系列文章

更多海思AI芯片方案学习笔记欢迎关注海思AI芯片方案学习。

BGR文件数据格式

小结

推荐阅读

目录