Improved Target Detection With YOLOv8 for GAN Augmented Polarimetric Images Using MIRNet Denoising Model

被引:0
作者
Dey, Jaydeep [1 ]
Anandan, P. [1 ]
Rajagopal, Sonaa [1 ]
Mani, Muralikrishnan [1 ]
机构
[1] Vellore Inst Technol, Sch Comp Sci & Engn, Chennai 600127, Tamil Nadu, India
关键词
Generative adversarial network (GAN); data augmentation; MIRNet; convolutional neural network (CNN); polarimetric images; polarization; denoising; peak signal to noise ratio (PSNR); structural similarity index (SSIM); you only look once (YOLO);
D O I
10.1109/ACCESS.2024.3496523
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Polarized images, which record the polarization characteristics of light, are becoming increasingly important in a variety of applications like remote sensing, medical imaging, and target detection. Their ability to offer additional information beyond traditional intensity-based images makes them valuable in situations where conventional imaging methods are lacking. However, the use of polarized images for tasks such as target detection presents challenges due to the limited availability of datasets, resulting in subpar performance in deep learning algorithms. Traditional methods for improving the quality of polarized images often involve noise reduction techniques, but these approaches may not fully exploit on the potential of deep learning algorithms due to limited dataset access. To get over this restriction and improve the performance of deep learning models on polarised images, new approaches are required. In this study, a new approach is proposed to address the challenges linked with polarized images by harnessing the capabilities of GAN and deep learning models. Specifically, the MIRNet CNN algorithm is utilized to denoise enhanced polarized datasets produced by GANs. By training the deep learning model on these enhanced datasets, the aim is to boost the performance of subsequent tasks like target detection. The study demonstrates the efficacy and efficiency of this novel approach for bettering the polarised image performance of deep learning models, particularly the MIRNet and YOLOv8 models. Through the use of GAN-generated enhanced datasets, there is a notable enhancement in the accuracy of target detection utilizing YOLOv8. This highlights the potential of this approach not only in target detection but also in various other fields that rely on precise object detection and image denoising utilizing polarized images.
引用
收藏
页码:166885 / 166910
页数:26
相关论文
共 28 条
[1]   FULLY POLARIMETRIC SAR IMAGE DESPECKLING USING DEEP NEURAL NETWROK [J].
Aghababaei, Hossein ;
Vitale, Sergio ;
Zamani, Roghayeh ;
Ferraioli, Giampaolo .
2022 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2022), 2022, :1488-1491
[2]   GAN-Based Data Augmentation with Vehicle Color Changes to Train a Vehicle Detection CNN [J].
Ayub, Aroona ;
Kim, HyungWon .
ELECTRONICS, 2024, 13 (07)
[3]   A new multimodal RGB and polarimetric image dataset for road scenes analysis [J].
Blin, Rachel ;
Ainouz, Samia ;
Canu, Stephane ;
Meriaudeau, Fabrice .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW 2020), 2020, :867-876
[4]  
Cai YH, 2021, ADV NEUR IN, V34
[5]   Towards Universal GAN Image Detection [J].
Cozzolino, Davide ;
Gragnaniello, Diego ;
Poggi, Giovanni ;
Verdoliva, Luisa .
2021 INTERNATIONAL CONFERENCE ON VISUAL COMMUNICATIONS AND IMAGE PROCESSING (VCIP), 2021,
[6]  
Gong P., 2020, J. Inf. Hiding Privacy Protection, V2, P155
[7]   Polarimetric image denoising on small datasets using deep transfer learning [J].
Hu, Haofeng ;
Jin, Huifeng ;
Liu, Hedong ;
Li, Xiaobo ;
Cheng, Zhenzhou ;
Liu, Tiegen ;
Zhai, Jingsheng .
OPTICS AND LASER TECHNOLOGY, 2023, 166
[8]  
Jiayuan Kong, 2021, 2021 13th International Conference on Advanced Infocomm Technology (ICAIT), P215, DOI 10.1109/ICAIT52638.2021.9701974
[9]   Impact of Traditional and Embedded Image Denoising on CNN-Based Deep Learning [J].
Kaur, Roopdeep ;
Karmakar, Gour ;
Imran, Muhammad .
APPLIED SCIENCES-BASEL, 2023, 13 (20)
[10]   GAN-Based Synthetic Data Augmentation for Infrared Small Target Detection [J].
Kim, Jun-Hyung ;
Hwang, Youngbae .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60