arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.11279cs.CVeess.IV
收藏

一种端到端级联图像去噪与目标检测神经网络

An End-to-End Cascaded Image Deraining and Object Detection Neural Network

University of Chinese Academy of Sciences(中国科学院大学)

Kaige Wang, Tianming Wang, Jianchuang Qu, Huatao Jiang, Qing Li, Lin Chang

摘要:

虽然近年来基于深度学习的图像去伪方法取得了很大的进展,但在现实世界中的应用中存在两个主要缺点。首先,以雨水去除为代表的低级视觉任务和以目标检测为代表的高级视觉任务之间的差距是显著的,低级视觉任务几乎不能对高级视觉任务做出贡献。第二,需要提高去噪数据集的质量。事实上,许多基线中的雨线与真实雨线有很大的差距,并且去噪数据集图像的分辨率通常不理想。同时,对于低级视觉任务和高级视觉任务,很少有共同的数据集。在本文中,我们探讨了低层次视觉任务与高层次视觉任务的结合。具体而言,我们提出了一个端到端的目标检测网络,以减少降雨的影响,它由两个级联的网络,一个改进的图像去噪网络和目标检测网络,分别。我们还设计了损失函数的组件,以适应不同的子网络的特性。然后,我们提出了一个基于KITTI数据集的数据集,用于降雨去除和目标检测,在此基础上,我们的网络超越了最先进的技术,在指标上有了显着的改进。此外,我们提出的网络在自动驾驶车辆收集的驾驶视频上进行了测量,并在除雨和目标检测方面显示出积极的结果。

英文摘要:

While the deep learning-based image deraining methods have made great progress in recent years, there are two major shortcomings in their application in real-world situations. Firstly, the gap between the low-level vision task represented by rain removal and the high-level vision task represented by object detection is significant, and the low-level vision task can hardly contribute to the high-level vision task. Secondly, the quality of the deraining dataset needs to be improved. In fact, the rain lines in many baselines have a large gap with the real rain lines, and the resolution of the deraining dataset images is generally not ideally. Meanwhile, there are few common datasets for both the low-level vision task and the high-level vision task. In this paper, we explore the combination of the low-level vision task with the high-level vision task. Specifically, we propose an end-to-end object detection network for reducing the impact of rainfall, which consists of two cascaded networks, an improved image deraining network and an object detection network, respectively. We also design the components of the loss function to accommodate the characteristics of the different sub-networks. We then propose a dataset based on the KITTI dataset for rainfall removal and object detection, on which our network surpasses the state-of-the-art with a significant improvement in metrics. Besides, our proposed network is measured on driving videos collected by self-driving vehicles and shows positive results for rain removal and object detection.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑