From 526e58d0d1a2f8e640a4791f8df79b8150f75040 Mon Sep 17 00:00:00 2001
From: Alexey <AlexeyAB@users.noreply.github.com>
Date: Sun, 10 Jun 2018 11:48:38 +0000
Subject: [PATCH] Merge pull request #1017 from jing-vision/master
---
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