From d7286c273211ffeb1f56594f863d1ee9922be6d4 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 07 Nov 2013 00:09:41 +0000
Subject: [PATCH] Loading may or may not work. But probably.
---
src/network.c | 50 ++++++++++++++++++++++++++++++++++++++++++++++++--
1 files changed, 48 insertions(+), 2 deletions(-)
diff --git a/src/network.c b/src/network.c
index 0a74b63..53184d9 100644
--- a/src/network.c
+++ b/src/network.c
@@ -5,6 +5,15 @@
#include "convolutional_layer.h"
#include "maxpool_layer.h"
+network make_network(int n)
+{
+ network net;
+ net.n = n;
+ net.layers = calloc(net.n, sizeof(void *));
+ net.types = calloc(net.n, sizeof(LAYER_TYPE));
+ return net;
+}
+
void run_network(image input, network net)
{
int i;
@@ -84,9 +93,9 @@
}
}
-double *get_network_output(network net)
+
+double *get_network_output_layer(network net, int i)
{
- int i = net.n-1;
if(net.types[i] == CONVOLUTIONAL){
convolutional_layer layer = *(convolutional_layer *)net.layers[i];
return layer.output.data;
@@ -101,6 +110,43 @@
}
return 0;
}
+
+int get_network_output_size_layer(network net, int i)
+{
+ if(net.types[i] == CONVOLUTIONAL){
+ convolutional_layer layer = *(convolutional_layer *)net.layers[i];
+ return layer.output.h*layer.output.w*layer.output.c;
+ }
+ else if(net.types[i] == MAXPOOL){
+ maxpool_layer layer = *(maxpool_layer *)net.layers[i];
+ return layer.output.h*layer.output.w*layer.output.c;
+ }
+ else if(net.types[i] == CONNECTED){
+ connected_layer layer = *(connected_layer *)net.layers[i];
+ return layer.outputs;
+ }
+ return 0;
+}
+
+double *get_network_output(network net)
+{
+ int i = net.n-1;
+ return get_network_output_layer(net, i);
+}
+
+image get_network_image_layer(network net, int i)
+{
+ if(net.types[i] == CONVOLUTIONAL){
+ convolutional_layer layer = *(convolutional_layer *)net.layers[i];
+ return layer.output;
+ }
+ else if(net.types[i] == MAXPOOL){
+ maxpool_layer layer = *(maxpool_layer *)net.layers[i];
+ return layer.output;
+ }
+ return make_image(0,0,0);
+}
+
image get_network_image(network net)
{
int i;
--
Gitblit v1.10.0