[发明专利]一种基于图像识别技术的排污识别方法及装置

权利要求书

1.一种基于图像识别技术的排污识别方法，其特征在于，包括：

获取排污口周边的视频图像；

利用预先构建的排污口目标检测网络模型识别排污口在所述视频图像中的位置坐标，以获得排污口的图像；

根据所述排污口的图像进行排污口排污动态识别。

2.根据权利要求1所述的排污识别方法，其特征在于，预先构建排污口目标检测网络模型的步骤包括：

获取所述目标检测网络模型所采用的网络以及训练参数；

将标注后的目标检测样本数据训练根据所述训练参数所述目标检测网络模型。

3.根据权利要求1所述的排污识别方法，其特征在于，所述根据所述排污口的图像进行排污口排污动态识别，包括：

根据所述排污口的图像获得对应的差值图像；

根据所述差值图像识别所述排污口是否在排污；

根据所述排污口的图像进行颜色区域分割；

根据分割的颜色区域进行污水判断；

所述根据所述差值图像识别所述排污口是否在排污，包括：

计算所述排污口的图像中颜色变化区域的像素面积；

比较所述颜色变化区域的像素面积与预设的第一阈值的大小，如果所述颜色变化区域的像素面积大于预设的第一阈值，则所述排污口正在排污。

4.根据权利要求3所述的排污识别方法，其特征在于，所述根据分割的颜色区域进行污水判断，包括：

计算所述排污区域的像素面积；

比较所述排污区域的像素面积与预设的第二阈值的大小，如果大于第二阈值，则所述排污口排出的是污水。

5.一种基于图像识别技术的排污识别装置，其特征在于，包括：

视频获取单元，用于获取排污口周边的视频图像；

位置坐标识别单元，用于利用预先构建的排污口目标检测网络模型识别排污口在所述视频图像中的位置坐标，以获得排污口的图像；

排污动态识别单元，用于根据所述排污口的图像进行排污口排污动态识别。

6.根据权利要求5所述的排污识别装置，其特征在于，预先构建排污口目标检测网络模型的步骤包括：

获取所述目标检测网络模型所采用的网络以及训练参数；

将标注后的目标检测样本数据训练根据所述训练参数所述目标检测网络模型。

7.根据权利要求5所述的排污识别装置，其特征在于，所述排污动态识别单元，包括：

差值图像获取模块，用于根据所述排污口的图像获得对应的差值图像；

排污判断模块，用于根据所述差值图像识别所述排污口是否在排污；

颜色分割模块，用于根据所述排污口的图像进行颜色区域分割；

污水判断模块，用于根据分割的颜色区域进行污水判断；

所述排污判断模块包括：

第一计算模块，用于计算所述排污口的图像中颜色变化区域的像素面积；

第一比较模块，用于比较所述颜色变化区域的像素面积与预设的第一阈值的大小，如果所述颜色变化区域的像素面积大于预设的第一阈值，则所述排污口正在排污。

8.根据权利要求7所述的排污识别装置，其特征在于，所述污水判断模块包括：

第二计算模块，用于计算所述排污区域的像素面积；

第二比较模块，用于比较所述排污区域的像素面积与预设的第二阈值的大小，如果大于第二阈值，则所述排污口排出的是污水。

9.一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现权利要求1至4中任一项所述基于图像识别技术的排污识别方法。

10.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，该计算机程序被处理器执行时实现权利要求1至4中任一项所述基于图像识别技术的排污识别方法。