From cc06817efa24f20811ef6b32143c6700a91c5f2a Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 11 Apr 2014 08:00:27 +0000
Subject: [PATCH] Attempt at visualizing ImageNet Features
---
src/network.c | 4 +++-
1 files changed, 3 insertions(+), 1 deletions(-)
diff --git a/src/network.c b/src/network.c
index e2c44b0..edae3c7 100644
--- a/src/network.c
+++ b/src/network.c
@@ -428,13 +428,14 @@
void visualize_network(network net)
{
+ image *prev = 0;
int i;
char buff[256];
for(i = 0; i < net.n; ++i){
sprintf(buff, "Layer %d", i);
if(net.types[i] == CONVOLUTIONAL){
convolutional_layer layer = *(convolutional_layer *)net.layers[i];
- visualize_convolutional_layer(layer, buff);
+ prev = visualize_convolutional_layer(layer, buff, prev);
}
}
}
@@ -506,3 +507,4 @@
return acc;
}
+
--
Gitblit v1.10.0