用DCGAN與VAE合成腦MR影像

本文做了兩個小實驗,分別利用兩個經典的生成模型(DCGAN與VAE)來合成腦影像;使用pytorch,全部代碼見github;實驗設備,NVIDIA 1080Ti; 訓練數據 腦影像使用Brats18 公開數據集的T2模態;提取axial面的240X240大小2d 切片,resize成128X128大小;保存爲.npy文件。 數據集示例 VAE 網絡結構 損失函數 第一項重建誤差,第二項KL lo
相關文章
相關標籤/搜索