官方代码:https://github.com/HuCaoFighting/Swin-Unet 目的:训练Swin-Unet分割肺部区域 官方数据集位置(可能下载不了):https://www.kaggle.com/datasets/nikhilpandey360/chest-xray-masks-and-labels CSDN免费下载数据集 实现效果: 输入原图 输出标签 此文中只用了整个数据集中的345张图像用来完成整个分割任务! 1...
1.1 检查test_vol.txt的内容是否是测试用的npz文件名称 测试集的npz文件 1.2 检查模型权重文件 2、修改部分代码 2.1 修改dataset_synapse.py slice_name=self.sample_list[idx].strip('\n')data_path=os.path.join(self.data_dir,slice_name+'.npz')data=np.load(data_path)image,label=data['image'],da...
这个项目使用主流的深度学习框架 Pytorch + UNet来实现,项目的特点是支持训练、分割算法特别轻量化、能够一键执行训练+预测,能够适应分割结构复杂的医学图像。项目提供完整的代码,包括训练 + 预测代码、一键执行脚本、训练好的分割模型权重 (当然也支持自己训练)、项目三方依赖库 (requirements.txt)、训练标注图片、待...