Table 2.

The detailed information of the U-net.

Input – 1 × 1 × 256
DownBlocks	DownBlock	Resnets – 1 × 32 × 256
	DownsampleBlock	Downsample – 1 × 32 × 128
		Resnets – 1 × 32 × 128
	AttnDownBlock	Downsample – 1 × 32 × 64
		Resnets & attentions – 1 × 64 × 64
MidBlock	Downsample – 1 × 64 × 32
	Resnets & attentions – 1 × 64 × 32
	Upsample – 1 × 64 × 64
UpBlocks	AttnUpBlock	Resnets & attentions – 1 × 32 × 64
		Upsample – 1 × 32 × 128
	UpsampleBlock	Resnets – 1 × 32 × 128
		Upsample – 1 × 32 × 256
	UpBlock	Resnets – 1 × 1 × 256
Output – 1 × 1 × 256