文档服务地址:http://47.92.0.57:3000/ 周报索引地址:http://47.92.0.57:3000/s/NruNXRYmV

Commit 7a9ce116 by 王肇一

channel num to 1

parent af74ed93
...@@ -102,7 +102,7 @@ if __name__ == "__main__": ...@@ -102,7 +102,7 @@ if __name__ == "__main__":
in_files = args.input in_files = args.input
out_files = get_output_filenames(args) 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)) logging.info("Loading model {}".format(args.model))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment