[发明专利]一种增强图像中植物色彩的方法和系统

权利要求书

1.一种增强图像中植物色彩的方法，其特征在于，所述方法包括：

获取待增强色彩的原始图像，利用SSD模型对原始图像中的植物进行检测与种类识别；

根据原始图像中检测到的植物区域中RGB三通道中的G/R、G/B的值，进一步判断该植物的颜色；以及，判断该植物是否为该原始图像的主体；

若该植物是原始图像的主体；

根据该植物的颜色特征，对原始图像中的植物进行差异化色彩增强。

2.根据权利要求1所述的方法，其特征在于，在利用SSD模型对原始图像中的植物进行检测与种类识别之前还包括对SSD模型进行训练，其训练方法包括：

搜集若干张含有植物的样本图像，按植物的叶子形状进行分类；

将样本图像输入SSD模型，设置不同尺寸的先验框，匹配不同外形的植物类别，生成对应的特征映射图；

特征映射图经过预测卷积层转换得到用于区分植物类别的预测框和对应的类别置信度；

通过类别损失函数和分类损失函数更新SSD模型参数，得到最优的SSD模型。

3.根据权利要求2所述的方法，其特征在于，对原始图像中的植物进行检测与种类识别包括

将原始图像输入到训练好的SSD模型中，通过类别置信度对边界框进行降序排列，若置信度高于预设值，则计算边界框与先验框的交并比；

利用非极大值抑制方法得到每种分类的最优预测框和所属的类别，输出识别结果。

4.根据权利要求1所述的方法，其特征在于，所述判断该植物的颜色包括通过SSD模型获取该植物叶子的边缘，描绘出该植物的边界区域，统计该边界区域内植物像素点的G/R_ave和G/B_ave，判断该植物的真实颜色。

5.根据权利要求1所述的方法，其特征在于，所述判断该植物是否为该原始图像的主体是根据该植物在原始图像中的位置或/和植物在原始图像中的像素占比来判断的。

6.根据权利要求5所述的方法，其特征在于，以该植物在原始图像中的位置来判断该植物是否为原始图像的主体包括：

设定原始图像中的某一点为图像中心坐标C(x，y)，以及，设定植物区域的某一点为植物中心坐标G(x₁，y₁)，原始图像的长宽分别为W、H，植物的心点与图像的中心点的距离为

若满足如下条件，则，判断该植物为原始图像的主体；

7.根据权利要求5所述的方法，其特征在于，以植物在原始图像中的像素占比来判断该植物是否为原始图像的主体包括：

计算原始图像的像素点的数量n，和计算植物区域在原始图像中的像素点的数量n₁，

若n₁≥0.4*n，则判断该植物为原始图像的主体。

8.根据权利要求1所述的方法，其特征在于，所述根据该植物的颜色特征，对原始图像中的植物进行差异化色彩增强包括：

获取植物在原始图像中输入像素值(R_in，G_in，B_in)，进行差异化色彩增强后的输出值为(R_out，G_out，B_out)，则增强色彩的过程为：

R_out＝k₁*R_in+b₁

G_out＝k₂*G_in+b₂

B_out＝k₃*B_in+b₃

其中，k₁，k₂，k₃代表RGB三通道的强度系数，b₁，b₂，b₃代表RGB三通道的偏移量。