Low-quality image object detection based on reinforcement learning adaptive enhancement

被引:6
作者
Ye, Jiongkai [1 ]
Wu, Yong [2 ]
Peng, Dongliang [1 ]
机构
[1] Hangzhou Dianzi Univ, Sch Automat, Hangzhou 310018, Peoples R China
[2] Zhejiang Inst Commun, Hangzhou 311112, Peoples R China
基金
中国国家自然科学基金;
关键词
Deep reinforcement learning; Image enhancement; Object detection; Multiple scenarios; Low -quality image;
D O I
10.1016/j.patrec.2024.04.019
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The task of object detection for low-quality images in multiple scenarios remains a challenging task. To alleviate this problem, a novel approach to improve the detection performance of the model on low-quality images using deep reinforcement learning is proposed. The approach differs from methods that involve data labeling and retraining, as the detection results of the model are utilized to formulate a reward function. This enables deep reinforcement learning to transform low-quality images into high-quality images through image enhancement, ultimately leading to an improvement in the model's recognition performance on low-quality images. An image enhancement tool chain (IETC) has been prepared, offering a flexible and lightweight algorithm selection for the Dueling Deep Q-Network-based Tool Selector (DDQN-TS). By adding a "Pass" option, Enable DDQN-TS to learn the relative relationship between performing and not performing image enhancement. To mitigate the potential inconsistency between the output of the detection model and the actual situation caused by negative samples, which can lead to incorrect decisions made by DDQN-TS, the introduction of a "thresh" parameter regulates the level of intervention in target detection by DIE. The experimental results demonstrate the effectiveness and portability of the DIE model in environments with noise, fog, and uneven brightness.
引用
收藏
页码:67 / 75
页数:9
相关论文
共 36 条
[1]  
[Anonymous], 2009, Proceedings of the 26th Annual International Conference on Machine Learning, DOI DOI 10.1145/1553374.1553422
[2]  
Ba J, 2014, ACS SYM SER
[3]   Large-Scale Machine Learning with Stochastic Gradient Descent [J].
Bottou, Leon .
COMPSTAT'2010: 19TH INTERNATIONAL CONFERENCE ON COMPUTATIONAL STATISTICS, 2010, :177-186
[4]   Albumentations: Fast and Flexible Image Augmentations [J].
Buslaev, Alexander ;
Iglovikov, Vladimir I. ;
Khvedchenya, Eugene ;
Parinov, Alex ;
Druzhinin, Mikhail ;
Kalinin, Alexandr A. .
INFORMATION, 2020, 11 (02)
[5]  
Chen Y., 2015, On Learning Optimized Reaction Diffusion Processes For Effective Image Restoration
[6]   The Pascal Visual Object Classes (VOC) Challenge [J].
Everingham, Mark ;
Van Gool, Luc ;
Williams, Christopher K. I. ;
Winn, John ;
Zisserman, Andrew .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2010, 88 (02) :303-338
[7]   Rich feature hierarchies for accurate object detection and semantic segmentation [J].
Girshick, Ross ;
Donahue, Jeff ;
Darrell, Trevor ;
Malik, Jitendra .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :580-587
[8]  
Hausknecht M., 2015, AAAI Fall Symposium Series
[9]   Single Image Haze Removal Using Dark Channel Prior [J].
He, Kaiming ;
Sun, Jian ;
Tang, Xiaoou .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2011, 33 (12) :2341-2353
[10]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778