[发明专利]一种基于视频的危险品检测方法及系统

说明书

本发明提供一种基于视频的危险品检测方法及系统，获取图像拍摄装置拍摄的视频流；对视频流进行分帧，并记录视频流中当前视频帧图像与前一视频帧图像之间的位移；当待识别包裹完全进入视频视野后，从对应的视频帧图像中提取出包含待识别包裹的小图，得到包裹框；对待识别包裹的小图进行危险品检测，得到危险品目标框；根据危险品目标框记录检测出的危险品目标信息。本发明充分分析了安检视频图像的特点，利用传统图像处理技术提取出视频图像中包裹小图，将包裹小图送入yolov4目标检测模型进行危险品检测，避免了直接对视频原图进行检测时的图像信息损失，有效提高了危险品检出率，降低了漏报率和误报率，提高了安检危险品检测的效果。

技术领域

本发明涉及危险品检测技术领域，特别是涉及一种基于视频的危险品检测方法及系统。

背景技术

伴随着我国经济发展的良好势头，我国新建的基础设施数量呈爆炸式增涨，新的机场、高铁站、地铁站如雨后春笋般涌现，人们工作和生活方面的出行需求量随之不断增长，关乎出行安全的危险品检测问题受到人们持续关注。对于目标检测技术，目前现有技术存在着以下四种缺陷：第一，对图像进行缩放处理会导致信息的丢失，增加漏报率和误报率；第二，对小目标检测能力不足，预测精度不高；第三，对每帧图像进行检测，影响实时性；第四，缺少跟踪模块，不能对危险品进行统计。

发明内容

鉴于以上所述现有技术的缺点，本发明的目的在于提供一种基于视频的危险品检测方法及系统，用于解决现有技术中危险品安全检测问题。

为实现上述目的及其他相关目的，本发明提供一种基于视频的危险品检测方法，包括以下步骤：

获取图像拍摄装置拍摄的视频流；

对所述视频流进行分帧，并记录所述视频流中当前视频帧图像与前一视频帧图像之间的位移；

当待识别包裹完全进入视频视野后，从对应的视频帧图像中提取出包含所述待识别包裹的小图，得到包裹框；

对所述待识别包裹的小图进行危险品检测，得到危险品目标框；

根据所述危险品目标框记录检测出的危险品目标信息；所述危险品目标信息包括：危险品目标编号、危险品类别、危险品目标框的坐标和危险品目标框的大小。

可选地，所述方法还包括：根据当前视频帧图像与前一视频帧图像之间的位移，更新所述包裹框和所述危险品目标框的坐标，并在所述视频流中显示更新后的包裹框和危险品目标框。

可选地，对所述待识别包裹的小图进行危险品检测的过程包括：

利用yolov4模型和tensorrt加速技术对所述待识别包裹的小图进行危险品检测，得到所述危险品目标框。

可选地，记录所述视频流中当前视频帧图像与前一视频帧图像之间的位移的过程包括：

获取分帧后的视频帧图像；

利用光流法对每一个视频帧图像进行跟踪，计算出当前视频帧图像与前一视频帧图像的位移。

本发明还提供一种基于视频的危险品检测系统，包括有：

视频读入模块，用于获取图像拍摄装置拍摄的视频流，并对所述视频流进行分帧；

目标跟踪模块，用于记录所述视频流中当前视频帧图像与前一视频帧图像之间的位移；

包裹提取模块，用于在待识别包裹完全进入视频视野后，从对应的视频帧图像中提取出包含所述待识别包裹的小图，得到包裹框；

目标检测模块，用于对所述待识别包裹的小图进行危险品检测，得到危险品目标框；