more writing stuff

Joseph Redmon

2015-09-23 a9e16d914a5e1247c4149d95afbe8f68ca846a53

 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);