From 3b864c2254752c252936c9cc14036991686d2c81 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 10 Jul 2015 23:36:08 +0000
Subject: [PATCH] width and stuff

---
 src/detection.c |    5 ++---
 1 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/detection.c b/src/detection.c
index dc3d9a4..94d3700 100644
--- a/src/detection.c
+++ b/src/detection.c
@@ -20,6 +20,7 @@
             j = (r*side + c) * elems;
             int class = max_index(box+j, classes);
             if(box[j+class] > 0.2){
+                int width = box[j+class]*5 + 1;
                 printf("%f %s\n", box[j+class], class_names[class]);
                 float red = get_color(0,class,classes);
                 float green = get_color(1,class,classes);
@@ -39,9 +40,7 @@
                 int right = (x+w/2)*im.w;
                 int top   = (y-h/2)*im.h;
                 int bot   = (y+h/2)*im.h;
-                draw_box(im, left, top, right, bot, red, green, blue);
-                draw_box(im, left+1, top+1, right+1, bot+1, red, green, blue);
-                draw_box(im, left-1, top-1, right-1, bot-1, red, green, blue);
+                draw_box_width(im, left, top, right, bot, width, red, green, blue);
             }
         }
     }

--
Gitblit v1.10.0