From: A method for improving semantic segmentation using thermographic images in infants
Parameter | Net | U-Net GAN | U-Net GAN + SA |
---|---|---|---|
Learning rate | 0.01 | 0.01 (generator) | 0.01 (generator) |
1e−4 (discriminator) | 1e−4 (discriminator) | ||
Batch size | 75 | 30 | 12 |
Epoch | 200 | 100 | 100 |