Table 5 Ablation study results of U-Net module.
Original U-Net | Self-attention | Multi-head attention | FID | KID | SSIM | MS-SSIM |
|---|---|---|---|---|---|---|
✓ | 18.58 | 14.72 | 0.4721 | 0.2801 | ||
✓ | ✓ | 17.45 | 12.43 | 0.4824 | 0.2816 | |
✓ | ✓ | 17.21 | 11.51 | 0.4883 | 0.2842 | |
✓ | ✓ | ✓ | 16.34 | 9.44 | 0.4912 | 0.2917 |