xujinzh / U-Net

医学图像分割经典深度学习网络Python代码实现

U-Net

医学图像分割经典深度学习网络 Python 代码实现: U-Net

代码介绍

代码是论文 "U-Net: Convolutional Networks for Biomedical Image Segmentation" 的 Python 实现。 本代码是基于 unet 的改进，主要是把代码从 TensorFlow 1 升级到 TensorFlow 2，并实现多GPU训练策略，同时增加代码的中文注释。

代码使用方法

1. git clone https://github.com/xujinzh/U-Net.git
2. python main.py --train ./data/membrane/train --test ./data/membrane/test --steps 100 --epochs 25

数据使用的是 ISBI Challenge: Segmentation of neuronal structures in EM stacks .

依赖包

• python3
• tensorflow2
• numpy
• skimage