From 6b38dcdce07b0c082334e7d9e10fe26bc440a347 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 08 Sep 2016 05:27:56 +0000
Subject: [PATCH] adding new tiny-yolo
---
src/region_layer.c | 6 ++----
1 files changed, 2 insertions(+), 4 deletions(-)
diff --git a/src/region_layer.c b/src/region_layer.c
index 2185ea5..5fe37c5 100644
--- a/src/region_layer.c
+++ b/src/region_layer.c
@@ -80,8 +80,8 @@
b.w = logistic_activate(x[index + 2]);
b.h = logistic_activate(x[index + 3]);
}
- //if(adjust && b.w < .01) b.w = .01;
- //if(adjust && b.h < .01) b.h = .01;
+ if(adjust && b.w < .01) b.w = .01;
+ if(adjust && b.h < .01) b.h = .01;
return b;
}
@@ -149,7 +149,6 @@
l.delta[index + 4] = l.noobject_scale * ((0 - l.output[index + 4]) * logistic_gradient(l.output[index + 4]));
if(best_iou > .5) l.delta[index + 4] = 0;
- /*
if(*(state.net.seen) < 6400){
box truth = {0};
truth.x = (i + .5)/l.w;
@@ -158,7 +157,6 @@
truth.h = .5;
delta_region_box(truth, l.output, index, i, j, l.w, l.h, l.delta, LOG, 1);
}
- */
}
}
}
--
Gitblit v1.10.0