Joseph Redmon
2015-09-23 a9e16d914a5e1247c4149d95afbe8f68ca846a53
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);