go-darknet/image.c

#include <darknet.h>

void fill_image_f32(image* im, int w, int h, int c, float* data) {
    int i;
    for (i = 0; i < w*h*c; i++) {
        im->data[i] = data[i];
    }
}

void set_data_f32_val(float* data, int index, float value) {
    data[index] = value;
}

void to_float_and_fill_image(image* im, int w, int h, uint8_t* data) {
    int x, y, idx_source;
    int pixel_count = w * h;
    int idx = 0;

    for (y = 0; y < h; y++) {
        for (x = 0; x < w; x++) {
            idx_source = (y*w + x) * 4;
            im->data[(pixel_count*0) + idx] = (float)data[idx_source] / 255;
            im->data[(pixel_count*1) + idx] = (float)data[idx_source+1] / 255;
            im->data[(pixel_count*2) + idx] = (float)data[idx_source+2] / 255;
            idx++;
        }
    }
}