Skip to content

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

Notifications You must be signed in to change notification settings

yukkizhang/U-net-simple

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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
也欢迎查看我的博客

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%