From aebe937710ced03d03f73ab23f410f29685655c1 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 11 Aug 2016 18:54:24 +0000
Subject: [PATCH] what do you even write here?
---
src/coco.c | 5 ++---
1 files changed, 2 insertions(+), 3 deletions(-)
diff --git a/src/coco.c b/src/coco.c
index ef1f12d..8ded951 100644
--- a/src/coco.c
+++ b/src/coco.c
@@ -89,7 +89,7 @@
avg_loss = avg_loss*.9 + loss*.1;
printf("%d: %f, %f avg, %f rate, %lf seconds, %d images\n", i, loss, avg_loss, get_current_rate(net), sec(clock()-time), i*imgs);
- if(i%1000==0){
+ if(i%1000==0 || (i < 1000 && i%100 == 0)){
char buff[256];
sprintf(buff, "%s/%s_%d.weights", backup_directory, base, i);
save_weights(net, buff);
@@ -348,9 +348,8 @@
convert_detections(predictions, l.classes, l.n, l.sqrt, l.side, 1, 1, thresh, probs, boxes, 0);
if (nms) do_nms_sort(boxes, probs, l.side*l.side*l.n, l.classes, nms);
draw_detections(im, l.side*l.side*l.n, thresh, boxes, probs, coco_classes, coco_labels, 80);
+ save_image(im, "prediction");
show_image(im, "predictions");
-
- show_image(sized, "resized");
free_image(im);
free_image(sized);
#ifdef OPENCV
--
Gitblit v1.10.0