From cf32e7e9b843560eb7ec3ed16e5b19f0f7156724 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@burninator.cs.washington.edu>
Date: Sat, 25 Jun 2016 23:12:00 +0000
Subject: [PATCH] colors
---
src/image.c | 9 +++++----
1 files changed, 5 insertions(+), 4 deletions(-)
diff --git a/src/image.c b/src/image.c
index 92833df..3bfa862 100644
--- a/src/image.c
+++ b/src/image.c
@@ -111,11 +111,11 @@
if(prob > thresh){
int width = pow(prob, 1./2.)*10+1;
width = 8;
- printf("%s: %.2f\n", names[class], prob);
- int offset = class*17 % classes;
- float red = get_color(0,offset,classes);
+ printf("%s: %.0f%%\n", names[class], prob*100);
+ int offset = class*1 % classes;
+ float red = get_color(2,offset,classes);
float green = get_color(1,offset,classes);
- float blue = get_color(2,offset,classes);
+ float blue = get_color(0,offset,classes);
float rgb[3];
rgb[0] = red;
rgb[1] = green;
@@ -365,6 +365,7 @@
image get_image_from_stream(CvCapture *cap)
{
IplImage* src = cvQueryFrame(cap);
+ if (!src) return make_empty_image(0,0,0);
image im = ipl_to_image(src);
rgbgr_image(im);
return im;
--
Gitblit v1.10.0