From 98813ad520e39541fbe3607ebb2f2519936955c7 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 29 Jul 2015 18:37:07 +0000
Subject: [PATCH] small bug in forward_network_gpu
---
src/image.c | 14 +++++++++++---
1 files changed, 11 insertions(+), 3 deletions(-)
diff --git a/src/image.c b/src/image.c
index ae8ca0a..5db93f1 100644
--- a/src/image.c
+++ b/src/image.c
@@ -188,7 +188,7 @@
int x,y,k;
image copy = copy_image(p);
constrain_image(copy);
- rgbgr_image(copy);
+ if(p.c == 3) rgbgr_image(copy);
//normalize_image(copy);
char buff[256];
@@ -849,8 +849,16 @@
void show_images(image *ims, int n, char *window)
{
image m = collapse_images_vert(ims, n);
- save_image(m, window);
- show_image(m, window);
+ int w = 448;
+ int h = ((float)m.h/m.w) * 448;
+ if(h > 896){
+ h = 896;
+ w = ((float)m.w/m.h) * 896;
+ }
+ image sized = resize_image(m, w, h);
+ save_image(sized, window);
+ show_image(sized, window);
+ free_image(sized);
free_image(m);
}
--
Gitblit v1.10.0