[发明专利]一种目标检测模型的建模方法和装置

说明书

本说明书实施例提供一种目标检测模型的建模方法和装置，其中方法包括：根据多个包含目标的通用图片样本进行深度学习模型训练，得到用于进行目标检测的基础检测模型，所述多个通用图片样本包括不同的背景信息；使用包含目标的多个背景图片样本，继续对所述基础检测模型进行迭代训练，得到目标检测模型，所述多个背景图片样本的背景信息是预定检测场景。

技术领域

本公开涉及监控技术领域，特别涉及一种目标检测模型的建模方法和装置。

背景技术

在视觉监控领域，经常需要在某种场景中检测出特定的目标。例如，可以在一个交叉口处设置监控摄像头，并由该摄像头采集的监控图像中检测出经过该交叉口的机动车、非机动车、行人等目标。目标检测的检测结果可以具有多种应用，比如，可以用于进行交叉口处的流量统计。

相关技术中，目标检测可以基于背景建模来实现。比如，仍以上述交叉口为例，可以使用该交叉口处的多幅图像的背景信息进行高斯背景建模，并将建好的背景模型与交叉口当前的监控图像进行比较，最终检测出图像中包含的前景目标。但是这种方法的检测效果不太好，有时会出现将前景的目标误识别为背景的情况，影响了目标检测的准确度。

发明内容

有鉴于此，本公开提供一种目标检测模型的建模方法和装置，以提高目标检测的准确度。

具体地，本说明书一个或多个实施例是通过如下技术方案实现的：

第一方面，提供一种目标检测模型的建模方法，所述方法包括：

根据多个包含目标的通用图片样本进行深度学习模型训练，得到用于进行目标检测的基础检测模型，所述多个通用图片样本包括不同的背景信息；

使用包含目标的多个背景图片样本，继续对所述基础检测模型进行迭代训练，得到目标检测模型，所述多个背景图片样本的背景信息是预定检测场景。

第二方面，提供一种目标检测模型的建模装置，所述装置包括：

基础模型训练模块，用于根据多个包含目标的通用图片样本进行深度学习模型训练，得到用于进行目标检测的基础检测模型，所述多个通用图片样本包括不同的背景信息；

背景样本调优模块，用于使用包含目标的多个背景图片样本，继续对所述基础检测模型进行迭代训练，得到目标检测模型，所述多个背景图片样本的背景信息是预定检测场景。

第三方面，提供一种目标检测模型的建模设备，所述设备包括存储器、处理器，以及存储在存储器上并可在处理器上运行的计算机指令，所述处理器执行指令时实现以下步骤：

根据多个包含目标的通用图片样本进行深度学习模型训练，得到用于进行目标检测的基础检测模型，所述多个通用图片样本包括不同的背景信息；

使用包含目标的多个背景图片样本，继续对所述基础检测模型进行迭代训练，得到目标检测模型，所述多个背景图片样本的背景信息是预定检测场景。

第四方面，提供一种计算机存储介质，所述存储介质内存储有计算机指令，所述计算机指令被处理器执行时实现如下的处理步骤：

根据多个包含目标的通用图片样本进行深度学习模型训练，得到用于进行目标检测的基础检测模型，所述多个通用图片样本包括不同的背景信息；

使用包含目标的多个背景图片样本，继续对所述基础检测模型进行迭代训练，得到目标检测模型，所述多个背景图片样本的背景信息是预定检测场景。