U-net

基于U-net的医学影像分割 / pytorch实现

Main Code

• main.py
  使用Unet模型对ISBI Challenge 2012的数据集（详见dataset）进行训练，并对结果进行预测。
• unet.py
  Unet模型的pytorch实现
• DataHelper.py
  此代码中包含函数功能有：
  1、对训练数据、测试数据的读取与矩阵化处理（在'main.py'被转成tensor）
  2、对预测结果的二值化处理

Operating environment

本次实验的环境如下：

• Ubuntu 16.04
  
• python 3.6
  
• cuda 9.0
  
• pytorch
  
  需要的第三方模块如下：
  
• numpy
  
• PIL
  
• tqdm
  
• skimage

Operation instructions

配置环境之后运行main.py

Sample

Reference

本次实验参考github有
https://github.com/JavisPeng/u_net_liver
https://github.com/zhixuhao/unet
也欢迎查看我的博客