From 83b2ae66e13ef2b83153699361127c73936a8763 Mon Sep 17 00:00:00 2001
From: vinjn <vinjn.z@gmail.com>
Date: Sun, 10 Jun 2018 01:45:39 +0000
Subject: [PATCH] Expose YOLODLL_API layer* get_network_layer(network* net, int i);
---
src/network.c | 5 +++++
src/network.h | 1 +
2 files changed, 6 insertions(+), 0 deletions(-)
diff --git a/src/network.c b/src/network.c
index 499bf0f..d135a29 100644
--- a/src/network.c
+++ b/src/network.c
@@ -493,6 +493,11 @@
return def;
}
+layer* get_network_layer(network* net, int i)
+{
+ return net->layers + i;
+}
+
image get_network_image(network net)
{
int i;
diff --git a/src/network.h b/src/network.h
index 4198c4b..66df6ff 100644
--- a/src/network.h
+++ b/src/network.h
@@ -134,6 +134,7 @@
void set_batch_network(network *net, int b);
int get_network_input_size(network net);
float get_network_cost(network net);
+YOLODLL_API layer* get_network_layer(network* net, int i);
YOLODLL_API detection *get_network_boxes(network *net, int w, int h, float thresh, float hier, int *map, int relative, int *num, int letter);
YOLODLL_API detection *make_network_boxes(network *net, float thresh, int *num);
YOLODLL_API void free_detections(detection *dets, int n);
--
Gitblit v1.10.0