From cff59ba1353b79ec3b69059ce1b4f191540616fd Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 16 Mar 2016 11:30:48 +0000
Subject: [PATCH] go updates

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

diff --git a/src/image.c b/src/image.c
index 0d714fe..ef76d45 100644
--- a/src/image.c
+++ b/src/image.c
@@ -142,15 +142,15 @@
     assert(im.w == im.h);
     int n, m;
     int c;
-        for(c = 0; c < im.c; ++c){
-            for(n = 0; n < im.w-1; ++n){
-                for(m = n + 1; m < im.w; ++m){
-                    float swap = im.data[m + im.w*(n + im.h*c)];
-                    im.data[m + im.w*(n + im.h*c)] = im.data[n + im.w*(m + im.h*c)];
-                    im.data[n + im.w*(m + im.h*c)] = swap;
-                }
+    for(c = 0; c < im.c; ++c){
+        for(n = 0; n < im.w-1; ++n){
+            for(m = n + 1; m < im.w; ++m){
+                float swap = im.data[m + im.w*(n + im.h*c)];
+                im.data[m + im.w*(n + im.h*c)] = im.data[n + im.w*(m + im.h*c)];
+                im.data[n + im.w*(m + im.h*c)] = swap;
             }
         }
+    }
 }
 
 void rotate_image_cw(image im, int times)
@@ -676,6 +676,17 @@
         }
     }
 
+    image binarize_image(image im)
+    {
+        image c = copy_image(im);
+        int i;
+        for(i = 0; i < im.w * im.h * im.c; ++i){
+            if(c.data[i] > .5) c.data[i] = 1;
+            else c.data[i] = 0;
+        }
+        return c;
+    }
+
     void saturate_image(image im, float sat)
     {
         rgb_to_hsv(im);
@@ -798,6 +809,8 @@
         image exp5 = copy_image(im);
         exposure_image(exp5, .5);
 
+        image bin = binarize_image(im);
+
 #ifdef GPU
         image r = resize_image(im, im.w, im.h);
         image black = make_image(im.w*2 + 3, im.h*2 + 3, 9);
@@ -817,7 +830,8 @@
         show_image(black2, "Recreate");
 #endif
 
-        show_image(im, "Original");
+        show_image(im,   "Original");
+        show_image(bin,  "Binary");
         show_image(gray, "Gray");
         show_image(sat2, "Saturation-2");
         show_image(sat5, "Saturation-.5");

--
Gitblit v1.10.0