[发明专利]货架图区域分割方法、装置、终端设备和计算机介质

权利要求书

1.一种货架图区域分割方法，包括：

获取原始图像；

将所述原始图像输入预先确定的分割模型，以得到目标图像；

对所述目标图像进行融合后处理，以得到结果图像；

将所述结果图像推送至具有显示功能的目标设备，以及控制所述目标设备显示所述结果图像。

2.根据权利要求1所述的方法，其中，所述预先确定的分割模型是通过如下步骤得到的：

获取样本集，其中，所述样本集中的样本包括样本原始图像和对应于所述样本原始图像的样本目标图像；

确定初始分割模型的结构以及初始化所述初始分割模型的参数；

从所述样本集中选取样本，以及执行以下训练步骤：

将选取的样本的样本原始图像输入至初始分割模型，得到所述选取的样本的目标图像；

将所述选取的样本的目标图像与对应的样本目标图像进行比较；

根据比较结果确定所述初始分割模型是否达到预设的优化目标；

响应于确定所述初始分割模型达到所述优化目标，将所述初始分割模型确定为预先确定的分割模型。

3.根据权利要求2所述的方法，其中，所述预先确定的分割模型包括编码器、双流自注意力模块和解码器，以及

所述将所述原始图像输入预先确定的分割模型，以得到目标图像，包括：

将所述原始图像输入所述编码器，以得到预处理图像特征；

将所述预处理图像特征输入所述双流自注意力模块，以得到后处理图像特征；

将所述后处理图像特征输入所述解码器，以得到所述目标图像。

4.根据权利要求3所述的方法，其中，所述双流自注意力模块包括空间自注意力模块、通道自注意力模块和融合模块；以及

所述将所述预处理图像特征输入所述双流自注意力模块，以得到后处理图像特征，包括：

将所述预处理图像特征输入所述空间自注意力模块，以得到预处理空间特征；

将所述预处理图像特征输入所述通道自注意力模块，以得到预处理通道特征；

将所述预处理空间特征和所述预处理通道特征输入所述融合模块，以得到所述后处理图像特征。

5.根据权利要求4所述的方法，其中，所述对所述目标图像进行融合后处理，以得到结果图像，包括：

从所述目标图像中提取缝隙区域集合，其中，所述缝隙区域集合中包括第一数目个缝隙区域；

对所述缝隙区域集合进行拟合处理，以得到分割线区域集合，其中，所述分割线区域集合包括第二数目个分割线区域；

基于所述结果图像和所述分割线区域集合，生成过程结果图像；

对所述过程结果图像进行过滤处理，以得到所述结果图像。

6.根据权利要求5所述的方法，其中，所述将所述选取的样本的目标图像与对应的样本目标图像进行比较，包括：

基于所述选取的样本的目标图像和对应的样本目标图像，利用下式，生成损失函数结果：

L＝αL_w+βL_A

其中，x为所述选取的样本的目标图像中的像素，M为所述选取的样本的目标图像中的像素总数，α、β为预先确定的控制参数，L_w为加权的交叉熵损失函数，w表征加权的交叉熵损失函数，L_A为边缘注意力损失函数，A表征边缘注意力损失函数，c为对应的样本目标图像中像素的类别标识，W_c为x对应的类别标识的权重，y_x为x对应的真实类别，p_x为分割模型针对x的预测概率。