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

权利要求书

1.一种基于视频的危险品检测方法，其特征在于，包括以下步骤：

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

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

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

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

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

2.根据权利要求1所述的基于视频的危险品检测方法，其特征在于，所述方法还包括：根据当前视频帧图像与前一视频帧图像之间的位移，更新所述包裹框和所述危险品目标框的坐标，并在所述视频流中显示更新后的包裹框和危险品目标框。

3.根据权利要求1或2所述的基于视频的危险品检测方法，其特征在于，对所述待识别包裹的小图进行危险品检测的过程包括：

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

4.根据权利要求1所述的基于视频的危险品检测方法，其特征在于，记录所述视频流中当前视频帧图像与前一视频帧图像之间的位移的过程包括：

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

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

5.一种基于视频的危险品检测系统，其特征在于，包括有：

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

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

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

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

跟踪处理模块，用于根据所述危险品目标框记录检测出的危险品目标信息，以及根据当前视频帧图像与前一视频帧图像之间的位移，更新所述包裹框和所述危险品目标框的坐标；所述危险品目标信息包括：危险品目标编号、危险品类别、危险品目标框的坐标和危险品目标框的大小；

信息展示模块，用于在所述视频流中显示更新后的包裹框和危险品目标框。

6.根据权利要求5所述的基于视频的危险品检测系统，其特征在于，所述目标检测模块对所述待识别包裹的小图进行危险品检测的过程包括：

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

7.根据权利要求5所述的基于视频的危险品检测系统，其特征在于，所述跟踪处理模块记录所述视频流中当前视频帧图像与前一视频帧图像之间的位移的过程包括：

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

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