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 | 8 +++++---
1 files changed, 5 insertions(+), 3 deletions(-)
diff --git a/src/writing.c b/src/writing.c
index ccfc844..02d3fcc 100644
--- a/src/writing.c
+++ b/src/writing.c
@@ -74,9 +74,7 @@
epoch = *net.seen/N;
char buff[256];
sprintf(buff, "%s/%s_%d.weights",backup_directory,base, epoch);
- //TODO
- if(get_current_batch(net)%10 == 0)
- save_weights(net, buff);
+ save_weights(net, buff);
}
}
}
@@ -104,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