From 1c0fd9bb4726f28b5ccf4491b8d108b00c884ec3 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 30 Oct 2014 06:26:41 +0000
Subject: [PATCH] im2col slightly faster

---
 src/image.c |   22 ++++++++++++++++++++--
 1 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/src/image.c b/src/image.c
index b25bf05..bf34e09 100644
--- a/src/image.c
+++ b/src/image.c
@@ -369,7 +369,6 @@
         // Will do a scaled image resize with the correct aspect ratio.
         outImg = resizeImage(croppedImg, newHeight, newWidth, 0);
         cvReleaseImage( &croppedImg );
-
     }
     else {
 
@@ -415,6 +414,25 @@
     return out;
 }
 
+image load_image_color(char *filename, int h, int w)
+{
+    IplImage* src = 0;
+    if( (src = cvLoadImage(filename, 1)) == 0 )
+    {
+        printf("Cannot load file image %s\n", filename);
+        exit(0);
+    }
+    if(h && w && (src->height != h || src->width != w)){
+        printf("Resized!\n");
+        IplImage *resized = resizeImage(src, h, w, 1);
+        cvReleaseImage(&src);
+        src = resized;
+    }
+    image out = ipl_to_image(src);
+    cvReleaseImage(&src);
+    return out;
+}
+
 image load_image(char *filename, int h, int w)
 {
     IplImage* src = 0;
@@ -720,7 +738,7 @@
 void show_images(image *ims, int n, char *window)
 {
     image m = collapse_images_vert(ims, n);
-    //save_image(m, window);
+    save_image(m, window);
     show_image(m, window);
     free_image(m);
 }

--
Gitblit v1.10.0