基于 MONAI 的 3D 图像分割任务2（Brain Tumour 和 SwinUNETR 训练）
文章目录
- 一、训练全部代码
- 二、训练过程可视化
- 三、总结
如标题所示，本节就基于 MONAI 库，采用Brain Tumour 的数据集，和 SwinUNETR 网络模型，搭建训练过程。
那么，本节就主要包含了以下部分：
- 获取数据过程
- 获取模型过程
- 构建损失和优化函数
- 训练、验证循环
- 保持优化后模型到本地
那么，娓娓道来。
一、训练全部代码
monai 提供了很多的算子，可以提供直接使用。可以立即为将原本较为复杂的、自己写的代码进行了封装。所以，就能剩下很多的时间，编写这部分代码。
但是，这就需要去参考官方的库文档，以增加对封装算子的理解。下面对训练部分所需要的代码，合并到一起。
import os
import shutil
import glob
import time
import torch
im