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/writing.c | 4 ++++
1 files changed, 4 insertions(+), 0 deletions(-)
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