GPU flag in Makefile

Joseph Redmon

2014-04-19 354b0cbdcb48a66de1af3aca9fc2f5687e46ee42

 src/tests.c

@@ -503,10 +503,11 @@
   IplImage *sized = cvCreateImage(cvSize(w,h), src->depth, src->nChannels);
   cvResize(src, sized, CV_INTER_LINEAR);
   image im = ipl_to_image(sized);
   normalize_array(im.data, im.h*im.w*im.c);
   //normalize_array(im.data, im.h*im.w*im.c);
   translate_image(im, -144);
   resize_network(net, im.h, im.w, im.c);
   forward_network(net, im.data);
   image out = get_network_image_layer(net, 6);
   image out = get_network_image(net);
   free_image(im);
   cvReleaseImage(&sized);
   return copy_image(out);
@@ -660,9 +661,8 @@
   cvWaitKey(0);
}

void features_VOC_image(char *image_file, char *image_dir, char *out_dir)
void features_VOC_image(char *image_file, char *image_dir, char *out_dir, int flip)
{
   int flip = 1;
   int interval = 4;
   int i,j;
   network net = parse_network_cfg("cfg/voc_imagenet.cfg");
@@ -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,12 +707,16 @@
         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);
      for(j = 0; j < out.c*out.h*out.w; ++j){
         if(j != 0)fprintf(fp, ",");
         fprintf(fp, "%g", out.data[j]);
         float o = out.data[j];
         if(o < 0) o = 0;
         fprintf(fp, "%g", o);
      }
      fprintf(fp, "\n");
      free_image(out);
@@ -787,10 +789,11 @@
   //test_vince();
   //test_full();
   //train_VOC();
   //features_VOC_image(argv[1], argv[2], argv[3]);
   //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_imagenet_topk("data/VOC2012.list");
   //visualize_cat();
   //flip_network();
   //test_visualize();