From a2f3644e87c11c222f86aa2aeb0e31d858ebb024 Mon Sep 17 00:00:00 2001
From: Alexey <AlexeyAB@users.noreply.github.com>
Date: Sat, 21 Apr 2018 20:56:46 +0000
Subject: [PATCH] Merge pull request #675 from IlyaOvodov/Flip_Fix
---
src/image.c | 16 ++++++++++++++--
1 files changed, 14 insertions(+), 2 deletions(-)
diff --git a/src/image.c b/src/image.c
index d29f0c1..61f760a 100644
--- a/src/image.c
+++ b/src/image.c
@@ -255,6 +255,12 @@
if (top < 0) top = 0;
if (bot > im.h - 1) bot = im.h - 1;
+ //int b_x_center = (left + right) / 2;
+ //int b_y_center = (top + bot) / 2;
+ //int b_width = right - left;
+ //int b_height = bot - top;
+ //sprintf(labelstr, "%d x %d - w: %d, h: %d", b_x_center, b_y_center, b_width, b_height);
+
draw_box_width(im, left, top, right, bot, width, red, green, blue);
if (alphabet) {
image label = get_label_v3(alphabet, labelstr, (im.h*.03));
@@ -393,6 +399,12 @@
if (top < 0) top = 0;
if (bot > show_img->height - 1) bot = show_img->height - 1;
+ //int b_x_center = (left + right) / 2;
+ //int b_y_center = (top + bot) / 2;
+ //int b_width = right - left;
+ //int b_height = bot - top;
+ //sprintf(labelstr, "%d x %d - w: %d, h: %d", b_x_center, b_y_center, b_width, b_height);
+
float const font_size = show_img->height / 1000.F;
CvPoint pt1, pt2, pt_text, pt_text_bg1, pt_text_bg2;
pt1.x = left;
@@ -418,7 +430,7 @@
black_color.val[0] = 0;
CvFont font;
cvInitFont(&font, CV_FONT_HERSHEY_SIMPLEX, font_size, font_size, 0, font_size * 3, 8);
- cvPutText(show_img, names[class_id], pt_text, &font, black_color);
+ cvPutText(show_img, labelstr, pt_text, &font, black_color);
}
}
}
@@ -560,7 +572,7 @@
cvPutText(img, char_buff, pt1, &font, CV_RGB(0, 0, 0));
cvShowImage("average loss", img);
int k = cvWaitKey(20);
- if (k == 's') cvSaveImage("chart.jpg", img, 0);
+ if (k == 's' || current_batch == (max_batches-1)) cvSaveImage("chart.jpg", img, 0);
}
#endif // OPENCV
--
Gitblit v1.10.0