对比性半监督学习的医学图像分割
该存储库实现了我们提交给MICCAI'21所需的组件
一些文件映射
attn_unet.py:骨干/基准代码注意unet
data_loader.py:包含腹腔数据集的数据生成器
generate_pseudolabels.py:一个实用程序文件,用于在无法快速运行的cpus非常慢的情况下生成和保存伪标签
inference.py:对一组测试图像进行推断
metrics.py:包含损失函数和要监控的指标
model.py:定义编码器,解码器,复合模型
train.py:运行网络的半监督CL + CR训练。 在开始的代码处更改配置变量并运行它(python train.py)
utils.py:一些功能实用程序
dataloaders / data_generator.py:腹腔数据集生成器的完全模块化实现。 可更改为其他数据集的Easilty
要求
2026-01-08 17:08:19
31KB
Python
1