From 23cb35e6c8eae8b59fab161036ae3f417a55c8db Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Fri, 30 Mar 2018 11:46:51 +0000
Subject: [PATCH] Changed small_object
---
src/cifar.c | 27 +++++++++++++++++++++++++--
1 files changed, 25 insertions(+), 2 deletions(-)
diff --git a/src/cifar.c b/src/cifar.c
index af1b4d6..4a27e79 100644
--- a/src/cifar.c
+++ b/src/cifar.c
@@ -137,8 +137,8 @@
axpy_cpu(10, 1, p, 1, pred, 1);
int index = max_index(pred, 10);
- int class = max_index(test.y.vals[i], 10);
- if(index == class) avg_acc += 1;
+ int class_id = max_index(test.y.vals[i], 10);
+ if(index == class_id) avg_acc += 1;
free_image(im);
printf("%4d: %.2f%%\n", i, 100.*avg_acc/(i+1));
}
@@ -166,6 +166,28 @@
free_data(test);
}
+void extract_cifar()
+{
+char *labels[] = {"airplane","automobile","bird","cat","deer","dog","frog","horse","ship","truck"};
+ int i;
+ data train = load_all_cifar10();
+ data test = load_cifar10_data("data/cifar/cifar-10-batches-bin/test_batch.bin");
+ for(i = 0; i < train.X.rows; ++i){
+ image im = float_to_image(32, 32, 3, train.X.vals[i]);
+ int class_id = max_index(train.y.vals[i], 10);
+ char buff[256];
+ sprintf(buff, "data/cifar/train/%d_%s",i,labels[class_id]);
+ save_image_png(im, buff);
+ }
+ for(i = 0; i < test.X.rows; ++i){
+ image im = float_to_image(32, 32, 3, test.X.vals[i]);
+ int class_id = max_index(test.y.vals[i], 10);
+ char buff[256];
+ sprintf(buff, "data/cifar/test/%d_%s",i,labels[class_id]);
+ save_image_png(im, buff);
+ }
+}
+
void test_cifar_csv(char *filename, char *weightfile)
{
network net = parse_network_cfg(filename);
@@ -243,6 +265,7 @@
char *cfg = argv[3];
char *weights = (argc > 4) ? argv[4] : 0;
if(0==strcmp(argv[2], "train")) train_cifar(cfg, weights);
+ else if(0==strcmp(argv[2], "extract")) extract_cifar();
else if(0==strcmp(argv[2], "distill")) train_cifar_distill(cfg, weights);
else if(0==strcmp(argv[2], "test")) test_cifar(cfg, weights);
else if(0==strcmp(argv[2], "multi")) test_cifar_multi(cfg, weights);
--
Gitblit v1.10.0