颜色分类leetcode pytorch bayesian unet:迁移到PyTorch

颜色分类leetcode BCNN这是贝叶斯卷积神经网络的PyTorch重新实现。 在这个项目中，我们假设了以下两种场景，尤其是医学成像。使用2D U-Net进行二维分割/回归。（例如，2D X射线、腹腔镜图像和CT切片）使用3D U-Net进行三维分割/回归。（例如，3D CT体积）这是以下作品的一部分。