A simple version for ProbUNet, the aim is to learn the paper of probUNet, so codes may be not easy to read, and there are still some bugs/ Please put your own dataset in ./data/ Present version dosen't support multi-gpu and model's inference, I will add quickly.