本文做了两个小实验,分别利用两个经典的生成模型(DCGAN与VAE)来合成脑影像;使用pytorch,全部代码见github;实验设备,NVIDIA 1080Ti;
训练数据
脑影像使用Brats18 公开数据集的T2模态;提取axial面的240X240大小2d 切片,resize成128X128大小;保存为.npy文件。
数据集示例
VAE
网络结构
损失函数
第一项重建误差,第二项KL loss;
本文做了两个小实验,分别利用两个经典的生成模型(DCGAN与VAE)来合成脑影像;使用pytorch,全部代码见github;实验设备,NVIDIA 1080Ti;
脑影像使用Brats18 公开数据集的T2模态;提取axial面的240X240大小2d 切片,resize成128X128大小;保存为.npy文件。
第一项重建误差,第二项KL loss;