【BraTS】Brain Tumor Segmentation 脑部肿瘤分割3（构建数据流）

往期回顾：

• 【BraTS】Brain Tumor Segmentation 脑部肿瘤分割1–数据篇
• 【BraTS】Brain Tumor Segmentation 脑部肿瘤分割2–UNet的复现

在上一篇网络复现中提到：

• 输入图像变成： 4 * 155 * 240 * 240，155层，每一层4个channel，每一个channel是240*240大小

在在接下来构建数据流时候，输出的图像部分，也就按照这个大小进行构建即可。

一、构建数据流

在数据篇中，了解到4个模态，每一个模态存储的都是一个3维数据。而上文网络输入是一个(batch_size, channel=4, width, height)，没有了Z轴的维度信息。所以，在构建数据的时候，需要按Z轴，将每一层单独拿出来，组成新