[发明专利]基于水色图像的水质评分方法、装置、设备及介质

权利要求书

1.一种基于水色图像的水质评分方法，其特征在于，所述方法包括：

获取水样图片，从所述水样图片中切割出目标水体图片；

将所述目标水体图片根据RGB三颜色通道划分为R值水样图片、G值水样图片和B值水样图片；

根据预设的颜色矩法分别获取所述R值水样图片、G值水样图片和B值水样图片的颜色矩，得到图像特征；

根据所述图像特征利用预设的水质评分模型，计算得到所述水样图片的水质评分。

2.如权利要求1所述的基于水色图像的水质评分方法，其特征在于，所述从所述水样图片中切割出目标水体图片，包括：

通过预设的智能识别网络判断所述水样图片中各个像素是否均为水体像素；

若所述水样图片中各个像素不均为水体像素，则裁剪所述水样图片，仅保留所述水样图片中的水体像素，得到标准水样图片；

若所述水样图片中各像素均为水体像素，则将所述水样图片作为标准水样图片；

将所述标准水样图片灰度化，得到灰度水样图片；

计算所述灰度水样图片中各个像素点与周边像素点的灰度变化梯度，将所述灰度变化梯度排序，得到最大变化梯度，并查找到最大灰度变化梯度对应像素点，得到目标像素点；

将所述目标像素点作为图片中心，以预设长宽数值对所述标准水样图片进行切割，得到所述目标水体图片。

3.如权利要求1所述的基于水色图像的水质评分方法，其特征在于，所述根据预设的颜色矩法分别获取所述R值水样图片、G值水样图片和B值水样图片的颜色矩，得到图像特征，包括：

分别获取所述R值水样图片、所述G值水样图片和所述B值水样图片的像素个数以及各个像素的颜色分量；

根据所述R值水样图片、所述G值水样图片和所述B值水样图片的像素个数以及各个像素的颜色分量利用预设的一至三阶颜色矩计算公式计算出所述R值水样图片的三个颜色矩，所述G值水样图片的三个颜色矩以及所述B值水样图片的三个颜色矩，得到9个颜色矩；

将9个所述颜色矩作为所述目标水体图片的分量特征，得到所述图像特征。

4.如权利要求3所述的基于水色图像的水质评分方法，其特征在于，所述一至三阶颜色矩计算公式分别为：

其中，μ_i、σ_i及S_i分别表示一阶颜色矩、二阶颜色矩及三阶颜色矩，p_i,j表示所述目标水体图片第j个像素的第i个颜色分量，N表示所述目标水体图片中的像素个数。

5.如权利要求1所述的基于水色图像的水质评分方法，其特征在于，所述根据所述图像特征利用预设的水质评分模型，计算得到所述水样图片的水质评分之前，所述方法还包括：

基于预设的决策树模型构建水质评分模型。

6.如权利要求5所述的基于水色图像的水质评分方法，其特征在于，所述基于预设的决策树模型构建水质评分模型，包括：

根据预设的决策树模型以及预设的指标函数构建初始水质评分模型；

获取预设的水质图片集，并将所述水质图片集分为训练集和检验集，并获取所述训练集中水质图片的水质评分；

获取所述训练集中水质图片的图片特征，根据所述图片特征和所述水质评分对所述初始水质评分模型进行训练，得到确定各个图片特征对应水质评分的初始水质评分模型；

利用所述检验集中的水质图片对所述初始水质模型进行检验，判断所述初始水质评分模型的并在检验通过时得到水质评分模型。

7.一种基于水色图像的水质评分装置，其特征在于，所述装置包括：

水样图片切割模块，用于获取水样图片，从所述水样图片中切割出目标水体图片；

颜色通道划分模块，用于将所述目标水体图片根据RGB三颜色通道划分为R值水样图片、G值水样图片和B值水样图片；

图像特征获取模块，用于根据预设的颜色矩法分别获取所述R值水样图片、G值水样图片和B值水样图片的颜色矩，得到图像特征；

水质评分计算模块，用于根据所述图像特征利用预设的水质评分模型，计算得到所述水样图片的水质评分。