From 372980d690f84aade1ebfd1a92750ed327ab1c8d Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 04 May 2015 18:29:21 +0000
Subject: [PATCH] i don't know what's going on anymore
---
src/network.c | 3 +++
1 files changed, 3 insertions(+), 0 deletions(-)
diff --git a/src/network.c b/src/network.c
index 5571076..3247a31 100644
--- a/src/network.c
+++ b/src/network.c
@@ -186,6 +186,9 @@
if(net.types[net.n-1] == COST){
return ((cost_layer *)net.layers[net.n-1])->output[0];
}
+ if(net.types[net.n-1] == DETECTION){
+ return ((detection_layer *)net.layers[net.n-1])->cost[0];
+ }
return 0;
}
--
Gitblit v1.10.0