[发明专利]雾天的目标检测方法、装置、电子设备以及存储介质

说明书

本公开提供了一种雾天的目标检测方法、装置、电子设备以及存储介质，涉及人工智能领域，具体涉及计算机视觉和深度学习技术，具体可用于智慧城市和智能交通场景下。具体实现方案为：从视频流中提取待检测图像，并获取待检测图像的多尺度的特征图；获取每个特征图对应的候选检测框的检测结果，其中，检测结果包括检测框内包括目标的置信度；根据多尺度的特征图中的一个特征图，获取待检测图像对应的雾浓度预测图；根据雾浓度预测图和置信度，对每个特征图的候选检测框进行过滤，以确定待检测图像的目标检测框。由此，能够提高雾天环境下目标检测的准确性，减少误检。

技术领域

本公开涉及人工智能领域，具体涉及计算机视觉和深度学习技术，具体可用于智慧城市和智能交通场景下。

背景技术

近年来，随着人工智能技术的发展，目标检测技术广泛用于智慧城市、智能交通、智能安防等场景。而由于室外天气变化多端，雾、霾等天气会造成能见度低，影响目标检测效果。

目前，相关技术在训练目标检测模型时，是将有雾数据和无雾数据进行混合训练。而有雾数据中，由于场景能见度低，目标通常比较模糊，对于模型而言学习难度较大，将此类目标加入模型训练，易使得该模型在雾天场景下产生更多误检。

发明内容

本公开提供了一种雾天的目标检测方法、装置、设备以及存储介质。

根据本公开的一方面，提供了一种雾天的目标检测方法，包括：

从视频流中提取待检测图像，并获取所述待检测图像的多尺度的特征图；

获取每个所述特征图对应的候选检测框的检测结果，其中，所述检测结果包括所述检测框内包括目标的置信度；

根据所述多尺度的特征图中的一个特征图，获取所述待检测图像对应的雾浓度预测图；

根据所述雾浓度预测图和所述置信度，对每个所述特征图的所述候选检测框进行过滤，以确定所述待检测图像的目标检测框。

根据本公开的另一方面，提供了一种雾天的目标检测的装置，包括：

提取模块，用于从视频流中提取待检测图像，并获取所述待检测图像的多尺度的特征图；

第一获取模块，用于获取每个所述特征图对应的候选检测框的检测结果，其中，所述检测结果包括所述检测框内包括目标的置信度；

第二获取模块，用于根据所述多尺度的特征图中的一个特征图，获取所述待检测图像对应的雾浓度预测图；

过滤模块，用于根据所述雾浓度预测图和所述置信度，对每个所述特征图的所述候选检测框进行过滤，以确定所述待检测图像的目标检测框。

根据本公开的另一方面，提供了一种电子设备，包括：

至少一个处理器；以及

与所述至少一个处理器通信连接的存储器；其中，

所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行雾天的目标检测方法。

根据本公开的另一方面，一种存储有计算机指令的非瞬时计算机可读存储介质，其中，所述计算机指令用于使所述计算机执行雾天的目标检测方法。

根据本公开的另一方面，一种计算机程序产品，包括计算机程序，所述计算机程序在被处理器执行时实现雾天的目标检测方法。

应当理解，本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征，也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。

附图说明

附图用于更好地理解本方案，不构成对本公开的限定。其中：