YOLO image结构体操作学习

image结构体定义

注意：里面有些代码跟原始的代码有点不一样，比如说class与classes是因为作者将yolo装成window vs下使用的，为了避免以后使用c++时候与类（class），就将源码里面的class统一改为classes。

typedef struct {
    int h;
    int w;
    int c;
    float *data;
} image;
注意：这里data的数据存储是
for c
    for h
        for w

        end
    end
end

image的操作

float get_color(int c, int x, int max)
{
  //这个函数是为了实现颜色的获取，使用的地方是在 draw_detections 中
    float ratio = ((float)x/max)*5;
    int i = floor(ratio);
    int j = ceil(ratio);
    ratio -= i;
    float r = (1-ratio) * colors[i][c] + ratio*colors[j][c];
    return r;
}

void flip_image(image a)
{
  //这个函数实现的是image额翻转（翻转的是根据a.w中心来进行的）
    int i,j,k;
    for(k = 0; k < a.c; ++k){
        for(i = 0; i < a.h; ++i){
            for(j = 0; j < a.w/2; ++j){
                int index = j + a.w*(i + a.h*(k));
                int flip = (a.w - j - 1) + a.w*(i + a.h*(k));
                //index 和 flip 是进行调换的位置
                float swap = a.data[flip];
                a.data[flip] = a.data[index];
                a.data[index] = swap;
            }
        }
    }
}
void draw_box(image a, int x1, int y1, int x2, int y2, float r, float g, float b)
{
    int i;
    //下面的操作是进行框边界的界定
    if(x1 < 0) x1 = 0;
    if(x1 >= a.w) x1 = a.w-1;
    if(x2 < 0) x2 = 0;
    if(x2 >= a.w) x2 = a.w-1;

    if(y1 < 0) y1 = 0;
    if(y1 >= a.h) y1 = a.h-1;
    if(y2 < 0) y2 = 0;
    if(y2 >= a.h) y2 = a.h-1;
    //画的是上下两条线
    for(i = x1; i <= x2; ++i){
        a.data[i + y1*a.w + 0*a.w*a.h] = r;
        a.data[i + y2*a.w + 0*a.w*a.h] = r;

        a.data[i + y1*a.w + 1*a.w*a.h] = g;
        a.data[i + y2*a.w + 1*a.w*a.h] = g;

        a.data[i + y1*a.w + 2*a.w*a.h] = b;
        a.data[i + y2*a.w + 2*a.w*a.h] = b;
    }
    //画的是左右两条线
    for(i = y1; i <= y2; ++i){
        a.data[x1 + i*a.w + 0*a.w*a.h] = r;
        a.data[x2 + i*a.w + 0*a.w*a.h] = r;

        a.data[x1 + i*a.w + 1*a.w*a.h] = g;
        a.data[x2 + i*a.w + 1*a.w*a.h] = g;

        a.data[x1 + i*a.w + 2*a.w*a.h] = b;
        a.data[x2 + i*a.w + 2*a.w*a.h] = b;
    }
}
void draw_box_width(image a, int x1, int y1, int x2, int y2, int w, float r, float g, float b)
{
  //这个函数是实现画宽度为w的矩形框，这个宽度是往里面画，不是往外画
    int i;
    for