README-zh.md

病理图像分割 Pytorch

基于Pytorch实现的病理图像分割，可作为语义分割的入门代码

数据集

• Brain MRI Segmentation
• Chest Xray Masks and Labels
• Skin Lesion Segmantation
• CHAOS
• Glomerulus

模型

• UNet
• UNet++
• Attention UNet
• R2U-Net
• Attention R2U-Net
• NFNPlus
• XNet

环境

• Pytorch 1.4.0
• Torchvision 0.5.0
• Python 3.7
• 一些其他的包请自行安装

准备

• 对于每份数据集，需要在其对应的类内修改其文件夹的绝对路径
• 每份数据集，需要生成train.json和val.json

使用

运行train.py，传入指定的命令行参数即可