骰子系数
matlab代码MRI分割
用于大脑异常分割的
U-Net
模型的实现.
有关原始源代码的更多信息,请查看作者编写的代码和代码。
数据集
用于训练该模型的数据集是
,可在
Kaggle
上获得。
它包含来自
110
名患者的数据。
数据由大小为
256x256x3
的
MRI
切片和相应的二进制掩码
256x256
组成。
患者的最小和最大切片数分别为
20
和
88。
训练
该网络使用
105
名患者进行训练,其余
5
名用于验证。
数据增强包括
-20
到
20
度之间的旋转、水平和垂直翻转。
损失是使用
计算的。
该模型在
GPU
上进行了
85
次训练。
检索具有最佳验证损失的权重以进行验证预测。
结果
平均验证准确率约为
88%。
下面切片中的绿色分割代表真实情况,红色分割代表模型的预测。
安装
要安装依赖项,请运行以下命令:
pip
install
-r
requirements.txt
如果使用
Conda,您还可以创建具有以下要求的环境:
conda
env
create
-f
environment.yml
默认情况下,环境名称为mri-segmentati
2023-03-23 19:04:16
34.27MB
系统开源
1