【开源库STBI基础】STBI图像库的理解

STBI图像数据和darknet中image数据类型之间的转换

前言

STBI图像数据和darknet中image数据类型之间的转换

image2stbi file: src/image.c function: save_image_options

    for(k = 0; k < im.c; ++k){                                   
        for(i = 0; i < im.w*im.h; ++i){                          
            data[i*im.c+k] = (unsigned char) (255*im.data[i + k*im.w*im.h]);
        }   
    }

其中，data是stbi数据，im是image数据类型，可以看出STBI数据存储方式和opencv类似，不过是RGB顺序，opencv中图像数据的存储方式如下图所示。

darknet中的image数据类型

typedef struct {                                                 
    int w;                                                       
    int h;                                                       
    int c;                                                       
    float *data;                                                 
} image;

stbi2image file: src/image.c function: load_image_stb

    for(k = 0; k < c; ++k){
        for(j = 0; j < h; ++j){                                  
            for(i = 0; i < w; ++i){                              
                int dst_index = i + w*j + w*h*k;                 
                int src_index = k + c*i + c*w*j;                 
                im.data[dst_index] = (float)data[src_index]/255.;
            }   
        }   
    }

darknet中image数据类型和opencv数据类型之间的转换

mat_to_image file: src/image_opencv.cpp load_image_cv

image mat_to_image(Mat m)
{
    int h = m.rows;
    int w = m.cols;
    int c = m.channels();
    image im = make_image(w, h, c);
    unsigned char *data = (unsigned char *)m.data;
    int step = m.step;
    int i, j, k;

    for(i = 0; i < h; ++i){
        for(k= 0; k < c; ++k){
            for(j = 0; j < w; ++j){
                im.data[k*w*h + i*w + j] = data[i*step + j*c + k]/255.;
            }
        }
    }
    rgbgr_image(im);
    return im;
}

image_to_mat file: src/image_opencv.cpp show_image_cv

Mat image_to_mat(image im)
{
    image copy = copy_image(im);
    constrain_image(copy);
    if(im.c == 3) rgbgr_image(copy);

    Mat m(cv::Size(im.w,im.h), CV_8UC(im.c));
    int x,y,c;

    int step = m.step;
    for(y = 0; y < im.h; ++y){
        for(x = 0; x < im.w; ++x){
            for(c= 0; c < im.c; ++c){
                float val = im.data[c*im.h*im.w + y*im.w + x];
                m.data[y*step + x*im.c + c] = (unsigned char)(val*255);
            }
        }
    }

    free_image(copy);
    return m;

}