This project is part of the work published to Neural Computing and Applications.
This code is an adaptation of Lee Ka Shing's code, available at https://github.com/kashing1999/pytorch-unet. Ka Shing's code was in turn adapted from a Pytorch implementation of the U-Net architecture made available by Alexandre Milesi. The original codebase is available at https://github.com/milesial/Pytorch-UNet/