From 40cc1046395385dd256012810866eba34904f034 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 28 Sep 2015 21:32:28 +0000
Subject: [PATCH] idk
---
src/yolo.c | 6 +++++-
1 files changed, 5 insertions(+), 1 deletions(-)
diff --git a/src/yolo.c b/src/yolo.c
index b2c89d8..4b241f3 100644
--- a/src/yolo.c
+++ b/src/yolo.c
@@ -65,7 +65,6 @@
if(weightfile){
load_weights(&net, weightfile);
}
- detection_layer layer = get_network_detection_layer(net);
int imgs = 128;
int i = *net.seen/imgs;
@@ -74,11 +73,16 @@
int N = plist->size;
paths = (char **)list_to_array(plist);
+ if(i*imgs > N*80){
+ net.layers[net.n-1].objectness = 0;
+ net.layers[net.n-1].joint = 1;
+ }
if(i*imgs > N*120){
net.layers[net.n-1].rescore = 1;
}
data train, buffer;
+ detection_layer layer = get_network_detection_layer(net);
int classes = layer.classes;
int background = layer.objectness;
int side = sqrt(get_detection_layer_locations(layer));
--
Gitblit v1.10.0