Acta Photonica Sinica, Volume. 51, Issue 12, 1210004(2022)
Infrared Ship Image Generation Algorithm Based on ISE-StyleGAN
Figures & Tables(14)
Fig. 8. Comparison of images generated by different generative adversarial network algorithms
Table 1. Experimental training parameters
View tableView in ArticleTable 1. Experimental training parameters
Parameter Value Learning rate TTUR Optimizer Adam Base size 64 Epoch 900
Table 2. PSNR index results of generative adversarial network model
View tableView in ArticleTable 2. PSNR index results of generative adversarial network model
Algorithm PSNR/dB Single scale Multiscale Small target Cloud and mist interference Night target DCGAN 13.254 7.363 6.784 10.436 9.473 CycleGAN 9.792 8.187 6.349 8.342 9.754 StyleGAN 8.835 7.829 4.382 4.993 8.438 ISE-StyleGAN 21.853 19.341 13.740 11.284 15.584
Table 3. MS-SSIM index results of generative adversarial network model
View tableView in ArticleTable 3. MS-SSIM index results of generative adversarial network model
Algorithm MS-SSIM/ % Single scale Multiscale Small target Cloud and mist interference Night target DCGAN 91.6 85.7 83.4 90.2 93.6 CycleGAN 83.8 88.1 84.7 79.0 89.2 StyleGAN 82.3 87.2 81.9 73.3 86.4 ISE-StyleGAN 97.9 94.5 89.3 92.7 98.5
Table 4. Data set composition of object detection test
View tableView in ArticleTable 4. Data set composition of object detection test
Number Algorithm Source data Conventionally generated data Algorithmic data generation Total 1 None 1 000 0 0 1 000 2 Conventional expansion 1 000 2 000 0 3 000 3 DCGAN 1 000 1 000 1 000 3 000 4 CycleGAN 1 000 1 000 1 000 3 000 5 StyleGAN 1 000 1 000 1 000 3 000 6 ISE-StyleGAN 1 000 1 000 1 000 3 000
Table 5. mAP value of target detection in each dataset
View tableView in ArticleTable 5. mAP value of target detection in each dataset
Number Algorithm YOLOv3 SSD Faster R⁃CNN Centernet 1 None 0.679 0.641 0.703 0.727 2 Conventional expansion 0.771 0.733 0.788 0.791 3 DCGAN 0.784 0.712 0.804 0.786 4 CycleGAN 0.788 0.745 0.801 0.787 5 StyleGAN 0.762 0.735 0.797 0.790 6 ISE-StyleGAN 0.833 0.812 0.822 0.817
Tools
Get Citation
Copy Citation Text
Haijun LI, Fancheng KONG, Junjie MU, Xiao LIU, Zhenbin DU, Yun LIN. Infrared Ship Image Generation Algorithm Based on ISE-StyleGAN[J]. Acta Photonica Sinica, 2022, 51(12): 1210004
Paper Information
Category:
Received: May. 18, 2022
Accepted: Jul. 6, 2022
Published Online: Feb. 6, 2023
The Author Email: KONG Fancheng (879445564@qq.com)
© Copyright 2018-2021 | Chinese Laser Press.
All Rights Reserved 沪ICP备15018463号-20