Figure 1.
Structure of CycleGAN
Figure 2.
Schematic diagram of SENet
Figure 3.
Diagram of the SE-Resnet module
Figure 4.
Schematic diagram of the attention mechanism for joining channels
Figure 5.
Improving the Generator Network Structure
Figure 6.
Comparison of Results before and after Adding Color Preservation Loss
Figure 7.
Loss Curve Variation Chart
Figure 8.
Loss Function Change Trend for the Original CycleGAN Model
Figure 9.
Loss Function Change Trend for the Improved CycleGAN Model
TABLE I.
Evaluation index table of ablation experiments
| Model | SSIM | PSNR |
|---|---|---|
| CycleGAN | 0.50 | 15.6 |
| CycleGAN+SENet | 0.61 | 16.7 |
| CycleGAN+improved loss function | 0.68 | 15.9 |
| CycleGAN+SENet+improved loss function. | 0.77 | 18.9 |
Figure 10.
Comparison of images generated by different models