From a9e16d914a5e1247c4149d95afbe8f68ca846a53 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 23 Sep 2015 00:34:48 +0000
Subject: [PATCH] more writing stuff
---
src/image.c | 10 ++++++++++
src/writing.c | 4 ++++
src/image.h | 2 ++
3 files changed, 16 insertions(+), 0 deletions(-)
diff --git a/src/image.c b/src/image.c
index 5209328..2811cb3 100644
--- a/src/image.c
+++ b/src/image.c
@@ -494,6 +494,16 @@
return gray;
}
+image threshold_image(image im, float thresh)
+{
+ int i;
+ image t = make_image(im.w, im.h, im.c);
+ for(i = 0; i < im.w*im.h*im.c; ++i){
+ t.data[i] = im.data[i]>0 ? 1 : 0;
+ }
+ return t;
+}
+
image blend_image(image fore, image back, float alpha)
{
assert(fore.w == back.w && fore.h == back.h && fore.c == back.c);
diff --git a/src/image.h b/src/image.h
index 27dc62a..e16d285 100644
--- a/src/image.h
+++ b/src/image.h
@@ -35,7 +35,9 @@
void hsv_to_rgb(image im);
void rgbgr_image(image im);
void constrain_image(image im);
+
image grayscale_image(image im);
+image threshold_image(image im, float thresh);
image collapse_image_layers(image source, int border);
image collapse_images_horz(image *ims, int n);
diff --git a/src/writing.c b/src/writing.c
index 7c23db1..02d3fcc 100644
--- a/src/writing.c
+++ b/src/writing.c
@@ -102,6 +102,10 @@
printf("%s: Predicted in %f seconds.\n", filename, sec(clock()-time));
image pred = get_network_image(net);
+ image t = threshold_image(pred, .2);
+ free_image(pred);
+ pred = t;
+
if (outfile) {
printf("Save image as %s.png (shape: %d %d)\n", outfile, pred.w, pred.h);
save_image(pred, outfile);
--
Gitblit v1.10.0