3DUnet实现3D医学影像的有效分割
最近涉及到了3D医学影像的分割，网络上相关的实现比力少，因此举行实现记录。1.设置代码环境
这里介绍一个很好的开源项目，git为： https://github.com/ellisdg/3DUnetCNN.git。
安装环境为：
nibabel>=4.0.1
numpy>=1.23.0
#torch>=1.12.0
monai>=1.2.0
scipy>=1.9.0
pandas>=1.4.3
nilearn>=0.9.1
pillow>=9.3.0
这里以Conda为例，很慢的话，可以-i 清湖镜像源：
conda create -n 3DUnet python=3.8
conda activate 3DUnet
git clone https://github.com/ellisdg/3DUnetCNN.git
cd 3DUnetCNN
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
2.设置数据集以及模型文件
这里以 examples/brats2020/brats2020_config.json的json设置文件为例。
https://i-blog.csdnimg.cn/direct/113de8c78a33470bbffa0d65b9a7de9d.png
在同级文件夹下创建我们的任务的设置文件。随后，对json文件中比力重要的参数举行说明
https://i-blog.csdnimg.cn/direct/cb71e87693e84461b8a9f96c14348eb7.png
json文件中，in_channels表现模型的输入通道，out_channels表现模型的输出通道数。
https://i-blog.csdnimg.cn/direct/8b246b2f6ab4413680274e91576e4fe2.png
dataset字典中，desired_shape就是颠末monai数据预处理处罚库处理处罚后的图片的WxHxC同一到128x128x128。labels就是数据集有几类，假如只有1类，那就只有0（背景）、1（前景之分）。该项目会将label转换成one hot编码。
https://i-blog.csdnimg.cn/direct/322fd90dedce4162ab0804ad73b7e249.png
接下来，比力重要的参数就是training_filenames，其中分为image和label，也就是图像以及其对应的标签。
https://i-blog.csdnimg.cn/direct/aab359cab4ab41899edc8bb57e8653e1.png
bratsvalidation_filenames就是测试集，只存图片image。
https://i-blog.csdnimg.cn/direct/dac3e6501fd74bcdacc99e05a24d3991.png
由于这是参考的 examples/brats2020/brats2020_config.json的大脑分割json设置文件，这里的数据集路径要更换为我们的，这里这里附上更改我们自己的数据集的脚本代码。
import json
import os.path
'''将自己的数据集进行划分并添加到json配置文件'''
#原json文件路径
filename = r'D:\jiedan\3DUnetCNN\examples\tooth_me\tooth_me_config.json'
#自己数据集的图片路径
my_data_dir = r'D:\jiedan\tooth_segmentation\image'
#自己数据集的label路径
my_data_label_dir = r'D:\jiedan\tooth_segmentation\label_32_pre'
#进行数据集划分
my_data_file = os.listdir(my_data_dir)
train_num, val_num = int(len(my_data_file) * 0.8), int(len(my_data_file) - len(my_data_file) * 0.8)
train_data_file = my_data_file[:train_num]
val_data_file = my_data_file
with open(filename, 'r') as opened_file:
data = json.load(opened_file)
#这里因为读取的所参考的examples/brats2020/brats2020_config.json
#该数据集的图片数远远大于我们自己的数据集，所以只要截取到和我们的数据集一致的长度就行
train_file = data["training_filenames"][:train_num]
val_file = data["bratsvalidation_filenames"][:val_num]
for index, file in enumerate(train_file[:train_num]):
file["image"] = os.path.join(my_data_dir, train_data_file)
file["label"] = os.path.join(my_data_label_dir, train_data_file.replace('.nii.gz', '.nii'))
for index_v, j in enumerate(val_file[:val_num]):
images_val = j['image']
j['image'] = os.path.join(my_data_dir, val_data_file)
#进行数据集的路径字典更新
data["training_filenames"] = train_file
data["bratsvalidation_filenames"] = val_file
with open(filename, 'w') as opened_file:
json.dump(data, opened_file, indent=4)# 使用indent参数格式化保存的JSON数据，以便更易于阅读
3.练习
下面是 练习的脚本。
python unet3d/scripts/train.py --config_filename
./examples/tooth_me/tooth_me_config.json
<config_filename>指向我们刚才处理处罚好的我们自己的数据集以及模型的json文件。
4.预测
下面是 预测的脚本。
python unet3d/scripts/train.py --config_filename
./examples/tooth_me/tooth_me_config.json
<config_filename>指向我们刚才处理处罚好的我们自己的数据集以及模型的json文件。
由于该git项目预测仅仅只是通过利用练习好的权重初始化的模型来输出预测图像，格式与输入图像划一，为nii.gz。
https://i-blog.csdnimg.cn/direct/208827d1475d4ff6b08b4beeea285fab.png
这个预测脚本 predict.py是没有衡量指标的计算的，比如Dice分数。
我们参考monai的官方文档的class monai.metrics.CumulativeIterationMetric类。
下面为官方文档利用说明：
dice_metric = DiceMetric(include_background=True, reduction="mean")
for val_data in val_loader:
val_outputs = model(val_data["img"])
val_outputs =
# compute metric for current iteration
dice_metric(y_pred=val_outputs, y=val_data["seg"])# callable to add metric to the buffer
# aggregate the final mean dice result
metric = dice_metric.aggregate().item()
# reset the status for next computation round
dice_metric.reset()
我们，首先定位到unet3d/scripts/predict.py，定位到 unet3d/predict/volumetric.py文件的volumetric_predictions函数。
def volumetric_predictions(model, dataloader, prediction_dir, activation=None, resample=False,
interpolation="trilinear", inferer=None):
output_filenames = list()
writer = NibabelWriter()
# 使用DiceMetric实例化metric对象
dice_metric = DiceMetric(include_background=True, reduction="mean")
......
with torch.no_grad():
for idx, item in enumerate(dataloader):
x = item["image"]
x = x.to(next(model.parameters()).device)# Set the input to the same device as the model parameters
.....
predictions = model(x)
batch_size = x.shape
for batch_idx in range(batch_size):
_prediction = predictions
_x = x
if resample:
_x = loader(os.path.abspath(_x.meta["filename_or_obj"]))
#在这里加上读取label的代码并转移到对应的device上
_label = loader(os.path.abspath(_x.meta["filename_or_obj"]).replace('image', 'label_32_pre').replace('nii.gz', 'nii'))
_label = _label.to(next(model.parameters()).device)# Set the input to the same device as the model parameters
_prediction = resampler(_prediction, _x)
#将模型预测的输出与加代码读取的label送进去
# compute metric for current iteration
dice_metric(y_pred=_prediction, y=_label)# callable to add metric to the buffe
writer.set_data_array(_prediction)
writer.set_metadata(_x.meta, resample=False)
out_filename = os.path.join(prediction_dir,
os.path.basename(_x.meta["filename_or_obj"]).split(".") + ".nii.gz")
writer.write(out_filename, verbose=True)
output_filenames.append(out_filename)
#最后求平均得到最终的Dice分数
# aggregate the final mean dice result
metric = dice_metric.aggregate().item()
return output_filenames
还有许多衡量的评价指标，可以参考monai的官方文档：
https://docs.monai.io/en/stable/metrics.html
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。
页:
[1]