[发明专利]图像分割方法和图像分割装置

权利要求书

1.一种图像分割方法，其特征在于，所述方法包括：

获取针对指定对象的多通道图像，其中，所述多通道图像至少包括红色通道和红外通道；

基于所述红色通道和所述红外通道确定针对所述指定对象的植被指数；

将所述植被指数作为一个增强通道，并且将所述增强通道至少与所述红外通道和所述红色通道进行拼接，得到增强特征图；

基于所述增强特征图对所述多通道图像进行图像分割。

2.根据权利要求1所述的方法，其特征在于，所述多通道图像还包括绿色通道和蓝色通道；

所述将所述植被指数作为一个增强通道，并且将所述增强通道至少与所述红外通道和所述红色通道进行拼接，得到增强特征图包括：

将所述植被指数作为一个增强通道，并且将所述增强通道与所述红外通道、所述红色通道、所述绿色通道和所述蓝色通道进行拼接，得到具有至少五个通道的增强特征图。

3.根据权利要求1所述的方法，其特征在于，在得到增强特征图之后，所述方法还包括：

对所述增强特征图进行归一化处理，得到归一化增强特征图；

所述基于所述增强特征图对所述多通道图像进行图像分割包括：基于所述归一化增强特征图对所述多通道图像进行图像分割。

4.根据权利要求1至3任一项所述的方法，其特征在于，所述基于所述增强特征图对所述多通道图像进行图像分割包括：

将所述增强特征图输入经训练的图像分割模型，得到针对所述多通道图像中各像素的分类结果，以对所述多通道图像进行图像分割。

5.根据权利要求4所述的方法，其特征在于，所述图像分割模型包括：

特征提取网络，所述特征提取网络包括多个成组的卷积层和池化层，所述特征提取网络的输入包括所述增强特征图，所述特征提取网络的输出包括针对所述多通道图像的多尺度特征；

特征融合网络，所述特征融合网络的输入包括针对所述多通道图像的多尺度特征，所述特征融合网络的输出包括所述多通道图像中各像素所属类别的概率。

6.根据权利要求5所述的方法，其特征在于，所述特征融合网络具体用于将针对所述多通道图像的多尺度特征以同一个像素的不同维度像素特征相加融合或者拼接融合中任意一种融合方式进行融合。

7.一种图像分割装置，其特征在于，所述装置包括：

多通道图像获取模块，用于获取针对指定对象的多通道图像，其中，所述多通道图像至少包括红色通道和红外通道；

植被指数确定模块，用于基于所述红色通道和所述红外通道确定针对所述指定对象的植被指数；

通道拼接模块，用于将所述植被指数作为一个增强通道，并且将所述增强通道至少与所述红外通道和所述红色通道进行拼接，得到增强特征图；

图像分割模块，用于基于所述增强特征图对所述多通道图像进行图像分割。

8.一种电子设备，包括：

一个或多个处理器；

存储装置，用于存储一个或多个程序，

其中，当所述一个或多个程序被所述一个或多个处理器执行时，使得所述一个或多个处理器执行根据权利要求1-6中任一项所述的方法。

9.一种计算机可读存储介质，其上存储有可执行指令，该指令被处理器执行时使处理器执行根据权利要求1-6中任一项所述的方法。

10.一种计算机程序产品，包括计算机程序，所述计算机程序被处理器执行时实现根据权利要求1-6中任一项所述的方法。