channel num to 1

7a9ce116 · 王肇一 · af74ed93 · 7a9ce116
Commit 7a9ce116 authored Jan 09, 2020 by 王肇一
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

predict.py predict.py +1 -1

No files found.
--- a/predict.py
+++ b/predict.py
@@ -102,7 +102,7 @@ if __name__ == "__main__":
    in_files = args.input
    out_files = get_output_filenames(args)

-    net = UNet(n_channels=3, n_classes=1)
+    net = UNet(n_channels=1, n_classes=1)

    logging.info("Loading model {}".format(args.model))