中国科学院重庆绿色智能技术研究院机构知识库

QQ客服 官方微博 反馈留言

	Multi-Task Learning for UAV Aerial Object Detection in Foggy Weather Condition
	Fang, Wenxuan1; Zhang, Guoqing1,2; Zheng, Yuhui1; Chen, Yuwen3
	2023-09-01
摘要	Adverse weather conditions such as haze and snowfall can degrade the quality of captured images and affect performance of drone detection. Therefore, it is challenging to locate and identify targets in adverse weather scenarios. In this paper, a novel model called Object Detection in a Foggy Condition with YOLO (ODFC-YOLO) is proposed, which performs image dehazing and object detection jointly by multi-task learning approach. Our model consists of a detection subnet and a dehazing subnet, which can be trained end-to-end to optimize both tasks. Specifically, we propose a Cross-Stage Partial Fusion Decoder (CSP-Decoder) in the dehazing subnet to recover clean features of encoder from complex weather conditions, thereby reducing the feature discrepancy between hazy and clean images, thus enhancing the feature consistency between different tasks. Additionally, to increase the feature modeling and representation capabilities of our network, we also propose an efficient Global Context Enhanced Extraction (GCEE) module to extract beneficial information from blurred images by constructing global feature context long-range dependencies. Furthermore, we propose a Correlation-Aware Aggregated Loss (CAALoss) to average noise patterns and tune gradient magnitudes across different tasks, accordingly implicitly enhancing data diversity and alleviating representation bias. Finally, we verify the advantages of our proposed model on both synthetic and real-world foggy datasets, and our ODFC-YOLO achieves the highest mAP on all datasets while achieving 36 FPS real-time detection speed.
关键词	UAV images object detection YOLO foggy weather condition
DOI	10.3390/rs15184617
发表期刊	REMOTE SENSING
卷号	15期号:18页码:18
通讯作者	Chen, Yuwen(chenyuwen@cigit.ac.cn)
收录类别	SCI
WOS记录号	WOS:001074532200001
语种	英语