[发明专利]皮肤创面图像分割方法、装置、设备及存储介质

权利要求书

1.一种皮肤创面图像分割方法，其特征在于，包括如下步骤：

获取包含有皮肤创面的待分割图像；

基于空间注意力机制和通道注意力机制对所述待分割图像进行边缘增强处理，得到第一边缘特征增强后的特征图，及对所述待分割图像进行空间增强处理，得到第一空间特征增强后的特征图；

对所述第一边缘特征增强后的特征图和所述第一空间特征增强后的特征图进行通道拼接，得到第一分割图像。

2.如权利要求1所述的皮肤创面图像分割方法，其特征在于，所述基于空间注意力机制和通道注意力机制对所述待分割图像进行边缘增强处理，得到第一边缘特征增强后的特征图，包括：

对所述待分割图像进行第一编码及第一最大池化得到第一特征图，对所述第一特征图进行第二编码及第二最大池化得到第二特征图，对所述第一特征图和所述第二特征图进行第一次边缘增强处理，得到第一边缘特征增强图；

对第m-1边缘特征增强图和第m+1特征图进行第m次边缘增强处理，得到第m边缘特征增强图；其中，第m+1特征图为对第m编码及第m最大池化得到的第m特征图进行第m+1编码得到的，m＝2,3,……N；

当m＝N时，得到的第N边缘特征增强图为所述第一边缘特征增强后的特征图。

3.如权利要求2所述的皮肤创面图像分割方法，其特征在于，所述基于空间注意力机制和通道注意力机制对所述待分割图像进行边缘增强处理，得到第一边缘特征增强后的特征图，包括：

基于空间注意力机制对所述第一特征图、第二特征图进行第一次边缘增强处理，得到第一边缘特征增强图；

基于空间注意力机制对所述第一边缘特征增强图、第三特征图进行第二次边缘增强处理，得到第二边缘特征增强图；

基于通道注意力机制对所述第二边缘特征增强图、第四特征图进行第三次边缘增强处理，得到第三边缘特征增强图；

基于通道注意力机制对所述第三边缘特征增强图、第五特征图进行第四次边缘增强处理，得到所述第一边缘特征增强后的特征图。

4.如权利要求2或3所述的皮肤创面图像分割方法，其特征在于，所述对第m-1边缘特征增强图和第m+1特征图进行第m次边缘增强处理，得到第m边缘特征增强图，包括：

对第m-1边缘特征增强图进行卷积和编码处理，得到第m-1边缘特征编码图，对第m+1特征图进行卷积和上采样处理，得到第m+1特征采样图；

对所述第m-1边缘特征编码图和第m+1特征采样图进行通道拼接，得到第m边缘特征拼接图；

基于空间注意力机制对所述第m边缘特征拼接图进行处理，得到空间权重；

所述空间权重通过激活函数后，与所述第m-1边缘特征编码图相乘得到所述第m边缘特征增强图。

5.如权利要求4所述的皮肤创面图像分割方法，其特征在于，所述基于空间注意力机制对所述第m边缘特征拼接图进行处理，得到空间权重，包括：

对所述第m边缘特征拼接图进行平均池化得到第一池化特征图，对所述第m边缘特征拼接图进行最大池化得到第二池化特征图；

对所述第一池化特征图和第二池化特征图进行通道拼接，得到拼接后的第m边缘特征池化图；

对所述第m边缘特征池化图进行卷积处理，得到空间权重。

6.如权利要求2或3所述的皮肤创面图像分割方法，其特征在于，所述对第m-1边缘特征增强图和第m+1特征图进行第m次边缘增强处理，得到第m边缘特征增强图，包括：

对第m-1边缘特征增强图进行卷积和编码处理，得到第m-1边缘特征编码图，对第m+1特征图进行卷积和上采样处理，得到第m+1特征采样图；

对所述第m-1边缘特征编码图和第m+1特征采样图进行通道拼接，得到第m边缘特征拼接图；

基于通道注意力机制对所述第m边缘特征拼接图进行处理，得到通道权重；

所述通道权重通过激活函数后，与所述第m-1边缘特征编码图相乘得到所述第m边缘特征增强图。