From 354b0cbdcb48a66de1af3aca9fc2f5687e46ee42 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sat, 19 Apr 2014 20:10:59 +0000
Subject: [PATCH] GPU flag in Makefile

---
 src/tests.c |   10 +++++-----
 1 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/tests.c b/src/tests.c
index 5b77dcc..851d781 100644
--- a/src/tests.c
+++ b/src/tests.c
@@ -672,8 +672,6 @@
 	if (flip)sprintf(out_path, "%s%d/%s_r.txt",out_dir, interval, image_file);
 	else sprintf(out_path, "%s%d/%s.txt",out_dir, interval, image_file);
 	printf("%s\n", image_file);
-	FILE *fp = fopen(out_path, "w");
-	if(fp == 0) file_error(out_path);
 
 	IplImage* src = 0;
 	if( (src = cvLoadImage(image_path,-1)) == 0 ) file_error(image_path);
@@ -709,6 +707,8 @@
 			ims[j+interval] = features_output_size(net, src, ex_h, ex_w);
 		}
 	}
+	FILE *fp = fopen(out_path, "w");
+	if(fp == 0) file_error(out_path);
 	for(i = 0; i < max_scale+interval; ++i){
 		image out = ims[i];
 		fprintf(fp, "%d, %d, %d\n",out.c, out.h, out.w);
@@ -789,9 +789,9 @@
 	//test_vince();
 	//test_full();
 	//train_VOC();
-	features_VOC_image(argv[1], argv[2], argv[3], 0);
-	features_VOC_image(argv[1], argv[2], argv[3], 1);
-	//features_VOC_image_size(argv[1], atoi(argv[2]), atoi(argv[3]));
+	//features_VOC_image(argv[1], argv[2], argv[3], 0);
+	//features_VOC_image(argv[1], argv[2], argv[3], 1);
+	features_VOC_image_size(argv[1], atoi(argv[2]), atoi(argv[3]));
 	//visualize_imagenet_features("data/assira/train.list");
 	//visualize_imagenet_topk("data/VOC2012.list");
 	//visualize_cat();

--
Gitblit v1.10.0