[发明专利]一种工地裸露土方的检测方法、装置及系统

权利要求书

1.一种工地裸露土方的检测方法，其特征在于，包括：

获取待检测工地视频图像帧；

将所述待检测工地视频图像帧进行数据增强，获取待检测工地视频图像帧集合；所述待检测工地视频图像帧集合包含至少两个样本图片；

将所述待检测工地视频图像帧集合进行目标检测，获取检测结果；

如果所述检测结果为存在裸露土方，则发送提醒消息。

2.根据权利要求1所述的工地裸露土方的检测方法，其特征在于，所述将所述待检测工地视频图像帧集合进行目标检测，获取检测结果的步骤，包括：

获取工地裸露土方置信度阈值；

将所述待检测工地视频图像帧集合中的样本一一进行目标检测，获取每个样本的图片的置信度和目标位置信息；

如果存在两个及两个以上所述样本图片的置信度都超过所述工地裸露土方置信度阈值，则检测结果为有效结果；

将所述有效结果进行归一化处理，并将所述两个及两个以上所述样本图片的置信度和所述目标位置信息进行加权平均，获取所述待检测工地视频图像帧的置信度及目标位置信息。

3.根据权利要求2所述的工地裸露土方的检测方法，其特征在于，所述将所述待检测工地视频图像帧集合进行目标检测，获取检测结果的步骤，还包括：

获取无效阈值；

根据所述工地裸露土方置信度阈值和所述无效阈值，将所述待检测工地视频图像帧集合中的样本一一进行目标检测，获取每个样本的图片的置信度和目标位置信息；

如果存在两个及两个以上所述样本图片的置信度都超过所述无效阈值，且最多一个大于所述工地裸露土方置信度阈值，则检测结果为误检结果；

将所述误检结果对应的待检测工地视频图像帧集合留存。

4.根据权利要求3所述的工地裸露土方的检测方法，其特征在于，所述数据增强至少包括：水平随机翻折，随机颜色扰动，随机降低和增加图片亮度模拟晚上和白天场景，随机增加高斯噪声，雨点和雪点模拟不同天气；随机缩放图片模拟远近不同尺度其中一种。

5.根据权利要求1-4中任意一项所述的工地裸露土方的检测方法，其特征在于，所述目标检测采用卷积神经网络进行数据处理；所述卷积神经网络包括：7个卷积层和3个最大池化层；所述卷积层的卷积核大小为3，最大池化层的卷积核大小为2。

6.一种工地裸露土方的检测装置，其特征在于，包括：

图像获取单元，用于获取待检测工地视频图像帧；

数据处理单元，用于将所述待检测工地视频图像帧进行数据增强，获取待检测工地视频图像帧集合；所述待检测工地视频图像帧集合包含至少两个样本图片；

目标检测单元，用于将所述待检测工地视频图像帧集合进行目标检测，获取检测结果；

消息发送单元，用于如果所述检测结果为存在裸露土方，则发送提醒消息。

7.根据权利要求6所述的工地裸露土方的检测装置，其特征在于，所述目标检测单元，还用于获取工地裸露土方置信度阈值；所述待检测工地视频图像帧集合中的样本一一进行目标检测，获取每个样本的图片的置信度和目标位置信息；如果存在两个及两个以上所述样本图片的置信度都超过所述工地裸露土方置信度阈值，则检测结果为有效结果；将所述有效结果进行归一化处理，并将所述两个及两个以上所述样本图片的置信度和所述目标位置信息进行加权平均，获取所述待检测工地视频图像帧的置信度及目标位置信息。

8.根据权利要求7所述的工地裸露土方的检测装置，其特征在于，所述目标检测单元，还用于获取无效阈值；根据所述工地裸露土方置信度阈值和所述无效阈值，将所述待检测工地视频图像帧集合中的样本一一进行目标检测，获取每个样本的图片的置信度和目标位置信息；如果存在两个及两个以上所述样本图片的置信度都超过所述无效阈值，且最多一个大于所述工地裸露土方置信度阈值，则检测结果为误检结果；将所述误检结果对应的待检测工地视频图像帧集合留存。