pytorch-unet来源：https://github.com/milesial/Pytorch-UNet前两天搞了一下图像分割，用了下unet。之前没怎么用过复现了一下18年的une pytorch 版本，记录学习一下1. 主函数if __name__ == '__main__':args = get_args()device = torch.device('cuda' if torch.c