detection prompt

This commit is contained in:
Joseph Redmon 2015-06-10 11:08:40 -07:00
parent a4485b8a66
commit f2ae18e825

View File

@ -219,11 +219,11 @@ void test_detection(char *cfgfile, char *weightfile)
clock_t time; clock_t time;
char filename[256]; char filename[256];
while(1){ while(1){
printf("Image Path: ");
fgets(filename, 256, stdin); fgets(filename, 256, stdin);
strtok(filename, "\n"); strtok(filename, "\n");
image im = load_image_color(filename,0,0); image im = load_image_color(filename,0,0);
image sized = resize_image(im, im_size, im_size); image sized = resize_image(im, im_size, im_size);
printf("%d %d %d\n", im.h, im.w, im.c);
float *X = sized.data; float *X = sized.data;
time=clock(); time=clock();
float *predictions = network_predict(net, X); float *predictions = network_predict(net, X);