C#/.net>> U-net>> 返回
项目作者： Czt1998

项目描述 ：
基于U-net的医学影像分割 / pytorch实现
高级语言： Python
项目地址: git://github.com/Czt1998/U-net.git
创建时间： 2019-11-17T15:24:57Z
项目社区：https://github.com/Czt1998/U-net
开源协议：
下载


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

也欢迎查看我的博客