From 451ef0a0a6b595bb8e4a945633659b4d31f0a372 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 20 Apr 2015 15:46:12 +0000
Subject: [PATCH] It's time, to du-du-du-du-DU-DU-DUEL!! https://www.youtube.com/watch?v=IVmtUK_1jh4
---
src/network.c | 12 ++++++++++++
1 files changed, 12 insertions(+), 0 deletions(-)
diff --git a/src/network.c b/src/network.c
index 75c9454..5571076 100644
--- a/src/network.c
+++ b/src/network.c
@@ -500,6 +500,18 @@
return get_network_input_size_layer(net, 0);
}
+detection_layer *get_network_detection_layer(network net)
+{
+ int i;
+ for(i = 0; i < net.n; ++i){
+ if(net.types[i] == DETECTION){
+ detection_layer *layer = (detection_layer *)net.layers[i];
+ return layer;
+ }
+ }
+ return 0;
+}
+
image get_network_image_layer(network net, int i)
{
if(net.types[i] == CONVOLUTIONAL){
--
Gitblit v1.10.0