channel 3

predict.py

@@ -102,7 +102,7 @@ if __name__ == "__main__":
     in_files = args.input
     out_files = get_output_filenames(args)
 
-    net = UNet(n_channels=1, n_classes=1)
+    net = UNet(n_channels=3, n_classes=1)
 
     logging.info("Loading model {}".format(args.model))
 

train.py

@@ -146,7 +146,7 @@ if __name__ == '__main__':
     #   - For 1 class and background, use n_classes=1
     #   - For 2 classes, use n_classes=1
     #   - For N > 2 classes, use n_classes=N
-    net = UNet(n_channels=1, n_classes=1)
+    net = UNet(n_channels=3, n_classes=1)
     logging.info(f'Network:\n'
                  f'\t{net.n_channels} input channels\n'
                  f'\t{net.n_classes} output channels (classes)\n'