From ec16ee612f8386d69176d02bf4147abaaec0fd00 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 17 Apr 2014 22:05:41 +0000
Subject: [PATCH] Old feature extraction method
---
src/tests.c | 19 +++++++++++--------
1 files changed, 11 insertions(+), 8 deletions(-)
diff --git a/src/tests.c b/src/tests.c
index 0319947..2789bed 100644
--- a/src/tests.c
+++ b/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,12 +661,11 @@
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");
+ network net = parse_network_cfg("cfg/voc_imagenet_nonorm.cfg");
char image_path[1024];
sprintf(image_path, "%s/%s",image_dir, image_file);
char out_path[1024];
@@ -714,7 +714,9 @@
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(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();
--
Gitblit v1.10.0