中国光学(中英文)2024,Vol.17Issue(5):1087-1097,11.DOI:10.37188/CO.2024-0011
夜间动物图像自监督学习增强与检测方法
Self-supervised learning enhancement and detection methods for nocturnal animal images
摘要
Abstract
In order to solve the problems of low image exposure,low contrast and difficulty of feature extrac-tion in real-time animal monitoring at night,we proposed a lightweight self-supervised deep neural network Zero-Denoise and an improved YOLOv8 model for image enhancement and accurate recognition of nocturn-al animal targets.The first stage of rapid enhancement was performed by lightweight PDCE-Net.A new lighting loss function was proposed,and the second stage of re-enhancement was carried out in PRED-Net based on the Retinex principle and the maximum entropy theory,using the original image and fast enhance-ment image corrected by the parameter adjustable Gamma.Then,the YOLOv8 model was improved to re-cognize the re-enhanced image.Finally,experimental analysis was conducted on the LOL dataset and the self-built animal dataset to verify the improvement of the Zero-Denoise network and YOLOv8 model for nocturnal animal target monitoring.The experimental results show that the PSNR,SSIM,and MAE indicat-ors of the Zero-Denoise network on the LOL dataset reach 28.53,0.76,and 26.15,respectively.Combined with the improved YOLOv8,the mAP value of the baseline model on the self-built animal dataset increases by 7.1%compared to YOLOv8.Zero-Denoise and improved YOLOv8 can achieve good quality images of nocturnal animal targets,which can be helpful in further study of accurate methods of monitoring these targets.关键词
夜间动物监测/低光增强/自监督学习/Retinex/低光去噪Key words
nocturnal animal detection/low-light enhancement/self-supervised learning/Retinex/low-light denoising分类
信息技术与安全科学引用本文复制引用
王驰,沈晨,黄庆,张国峰,卢汉,陈金波..夜间动物图像自监督学习增强与检测方法[J].中国光学(中英文),2024,17(5):1087-1097,11.基金项目
国家自然科学基金项目(No.62175144) (No.62175144)
北京市航空智能遥感装备工程技术研究中心开放基金课题(No.AIRSE20233) Supported by the National Natural Science Foundation of China(No.62175144) (No.AIRSE20233)
Beijing Engineering Research Center of Aerial Intelligent Remote Sensing Equipments Open Fund(No.AIRSE20233) (No.AIRSE20233)