From f996bd59a61338d8d51e2b19482d684f6dd04d0f Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 23 Sep 2015 21:16:48 +0000
Subject: [PATCH] more writing fixes
---
src/yolo.c | 6 ++++--
1 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/src/yolo.c b/src/yolo.c
index 9b229e2..b2c89d8 100644
--- a/src/yolo.c
+++ b/src/yolo.c
@@ -290,14 +290,16 @@
set_batch_network(&net, 1);
srand(2222222);
clock_t time;
- char input[256];
+ char buff[256];
+ char *input = buff;
while(1){
if(filename){
strncpy(input, filename, 256);
} else {
printf("Enter Image Path: ");
fflush(stdout);
- fgets(input, 256, stdin);
+ input = fgets(input, 256, stdin);
+ if(!input) return;
strtok(input, "\n");
}
image im = load_image_color(input,0,0);
--
Gitblit v1.10.0