From d1965bdb969920c85f72785ec6e1f3d7bda957de Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 14 Mar 2016 06:18:42 +0000
Subject: [PATCH] Go
---
src/tag.c | 11 +++++++----
1 files changed, 7 insertions(+), 4 deletions(-)
diff --git a/src/tag.c b/src/tag.c
index 8b63d31..a00a161 100644
--- a/src/tag.c
+++ b/src/tag.c
@@ -99,6 +99,7 @@
int indexes[10];
char buff[256];
char *input = buff;
+ int size = net.w;
while(1){
if(filename){
strncpy(input, filename, 256);
@@ -109,11 +110,12 @@
if(!input) return;
strtok(input, "\n");
}
- image im = load_image_color(input, net.w, net.h);
- //resize_network(&net, im.w, im.h);
- printf("%d %d\n", im.w, im.h);
+ image im = load_image_color(input, 0, 0);
+ image r = resize_min(im, size);
+ resize_network(&net, r.w, r.h);
+ printf("%d %d\n", r.w, r.h);
- float *X = im.data;
+ float *X = r.data;
time=clock();
float *predictions = network_predict(net, X);
top_predictions(net, 10, indexes);
@@ -122,6 +124,7 @@
int index = indexes[i];
printf("%.1f%%: %s\n", predictions[index]*100, names[index]);
}
+ free_image(r);
free_image(im);
if (filename) break;
}
--
Gitblit v1.10.0