From eff487ba3626a39e135d13929117e04bc4cf5823 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Sun, 10 Jun 2018 13:36:33 +0000
Subject: [PATCH] Minor fix
---
src/data.c | 2 +-
src/detector.c | 14 ++++++++++++--
2 files changed, 13 insertions(+), 3 deletions(-)
diff --git a/src/data.c b/src/data.c
index a685fb4..d92e307 100644
--- a/src/data.c
+++ b/src/data.c
@@ -342,7 +342,7 @@
system(buff);
continue;
}
- if (x < 0 || x > 1 || y < 0 || y > 1) {
+ if (x <= 0 || x > 1 || y <= 0 || y > 1) {
printf("\n Wrong annotation: x = %f, y = %f \n", x, y);
sprintf(buff, "echo %s \"Wrong annotation: x = %f, y = %f\" >> bad_label.list", labelpath, x, y);
system(buff);
diff --git a/src/detector.c b/src/detector.c
index 228647a..2a4c4a2 100644
--- a/src/detector.c
+++ b/src/detector.c
@@ -916,8 +916,18 @@
int num_labels = 0;
box_label *truth = read_boxes(labelpath, &num_labels);
//printf(" new path: %s \n", labelpath);
+ char buff[1024];
for (j = 0; j < num_labels; ++j)
{
+ if (truth[j].x > 1 || truth[j].x <= 0 || truth[j].y > 1 || truth[j].y <= 0 ||
+ truth[j].w > 1 || truth[j].w <= 0 || truth[j].h > 1 || truth[j].h <= 0)
+ {
+ printf("\n\nWrong label: %s - j = %d, x = %f, y = %f, width = %f, height = %f \n",
+ labelpath, j, truth[j].x, truth[j].y, truth[j].w, truth[j].h);
+ sprintf(buff, "echo \"Wrong label: %s - j = %d, x = %f, y = %f, width = %f, height = %f\" >> bad_label.list",
+ labelpath, j, truth[j].x, truth[j].y, truth[j].w, truth[j].h);
+ system(buff);
+ }
number_of_boxes++;
rel_width_height_array = realloc(rel_width_height_array, 2 * number_of_boxes * sizeof(float));
rel_width_height_array[number_of_boxes * 2 - 2] = truth[j].w * width;
@@ -1000,8 +1010,8 @@
float box_intersect = min_w*min_h;
float box_union = box_w*box_h + anchor_w*anchor_h - box_intersect;
float iou = box_intersect / box_union;
- if (iou > 1 || iou < 0 || box_w > width || box_h > height) {
- printf(" i = %d, box_w = %d, box_h = %d, anchor_w = %d, anchor_h = %d, iou = %f \n",
+ if (iou > 1 || iou < 0) { // || box_w > width || box_h > height) {
+ printf(" Wrong label: i = %d, box_w = %d, box_h = %d, anchor_w = %d, anchor_h = %d, iou = %f \n",
i, box_w, box_h, anchor_w, anchor_h, iou);
}
else avg_iou += iou;
--
Gitblit v1.10.0