mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
faster resize?
This commit is contained in:
50
src/image.c
50
src/image.c
@ -584,6 +584,51 @@ image resize_image(image im, int w, int h)
|
||||
return resized;
|
||||
}
|
||||
|
||||
image resize_image2(image im, int w, int h)
|
||||
{
|
||||
image resized = make_image(w, h, im.c);
|
||||
image part = make_image(w, im.h, im.c);
|
||||
int r, c, k;
|
||||
float w_scale = (float)(im.w - 1) / (w - 1);
|
||||
float h_scale = (float)(im.h - 1) / (h - 1);
|
||||
for(k = 0; k < im.c; ++k){
|
||||
for(r = 0; r < im.h; ++r){
|
||||
for(c = 0; c < w; ++c){
|
||||
float val = 0;
|
||||
if(c == w-1){
|
||||
val = get_pixel(im, im.w-1, r, k);
|
||||
} else {
|
||||
float sx = c*w_scale;
|
||||
int ix = (int) sx;
|
||||
float dx = sx - ix;
|
||||
val = (1 - dx) * get_pixel(im, ix, r, k) + dx * get_pixel(im, ix+1, r, k);
|
||||
}
|
||||
set_pixel(part, c, r, k, val);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
for(k = 0; k < im.c; ++k){
|
||||
for(r = 0; r < h; ++r){
|
||||
float sy = r*h_scale;
|
||||
int iy = (int) sy;
|
||||
float dy = sy - iy;
|
||||
for(c = 0; c < w; ++c){
|
||||
float val = (1-dy) * get_pixel(part, c, iy, k);
|
||||
set_pixel(resized, c, r, k, val);
|
||||
}
|
||||
if(r == h-1) continue;
|
||||
for(c = 0; c < w; ++c){
|
||||
float val = dy * get_pixel(part, c, iy+1, k);
|
||||
add_pixel(resized, c, r, k, val);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
free_image(part);
|
||||
return resized;
|
||||
}
|
||||
|
||||
void test_resize(char *filename)
|
||||
{
|
||||
image im = load_image(filename, 0,0, 3);
|
||||
@ -728,6 +773,11 @@ void set_pixel(image m, int x, int y, int c, float val)
|
||||
assert(x < m.w && y < m.h && c < m.c);
|
||||
m.data[c*m.h*m.w + y*m.w + x] = val;
|
||||
}
|
||||
void add_pixel(image m, int x, int y, int c, float val)
|
||||
{
|
||||
assert(x < m.w && y < m.h && c < m.c);
|
||||
m.data[c*m.h*m.w + y*m.w + x] += val;
|
||||
}
|
||||
|
||||
void print_image(image m)
|
||||
{
|
||||
|
Reference in New Issue
Block a user