SegTransVAE 建立在编码器-解码器架构之上,利用带有变分自动编码器 (VAE) 分支的变换器与网络一起重建输入图像并进行分割。据我们所知,这是第一个将 CNN、transformer 和 VAE 的成功结合起来的方法。对最近引入的各种数据集的评估表明,SegTransVAE 在 Dice Score 和 95%95%-Haudorff 距离方面优于以前的方法,...
SegTransVAE Here is the network architecture of SegTransVAE, a hybrid CNN-Transformer-VAE for medical image segmentation Requirements python 3.8 pytorch 1.8.0 monai 0.6 pytorch lightning nibabel itk Data After downloading BraTS 2021 dataset, create a json file, which contains the path to images ...