Swin-Unet on 2D images

[Gnkgo]

Is it possible to use Swin-Unet on 2D images for different types of segmentation tasks? And if so, how should I proceed?

[GZ-YourZY]

Is it possible to use Swin-Unet on 2D images for different types of segmentation tasks? And if so, how should I proceed?

Do you have a solution? I also need the 2D version of Swin-UNet for experiments.

[Gnkgo]

Hey, I managed to run the code but all my predictions were plain black. I tried to allocate one single image to the .nii.gz file, but something is still wrong.

I changed the model for my segmentation task, which are better for 2D images. Maybe try one of them:
nnUNet, UNet++, Finetune-SAM, SAAB

[18834114892]

嘿，我设法运行了代码，但我所有的预测都是纯黑色的。我尝试将一个图像分配给 .nii.gz 文件，但仍然有问题。

我更改了分割任务的模型，该模型更适合 2D 图像。也许可以尝试其中之一：nnUNet、UNet++、Finetune-SAM、SAAB

您好，我想请问下，3D分类和2D分类是只有网络模型不同，train文件和test文件需要修改吗