[发明专利]压板状态确定方法、装置、计算机设备和可读存储介质

权利要求书

1.一种压板状态确定方法，其特征在于，所述方法包括：

采集压板原始图像，所述压板原始图像中包括至少一个单体压板；

基于所述压板原始图像，生成对应的掩码图像；

根据所述掩码图像，确定所述压板原始图像中各单体压板的边缘直线；

基于各所述边缘直线，确定所述压板原始图像中各单体压板的压板状态。

2.根据权利要求1所述的方法，其特征在于，所述根据所述掩码图像，确定单体压板的边缘直线，包括：

对所述掩码图像做连通域提取，确定对应各单体压板的单体掩码图像；

对各所述单体掩码图像进行边缘直线检测，确定各所述单体压板的边缘直线。

3.根据权利要求2所述的方法，其特征在于，所述对所述掩码图像做连通域提取，确定对应各单体压板的单体掩码图像，包括：

获取掩码图像中各像素点的像素值以及各所述像素点的坐标位置；

根据各所述像素值以及各所述坐标位置，确定掩码图像中相连接的像素点；

基于相连通的像素点，确定连通域，得到对应各所述单体压板的单体掩码图像。

4.根据权利要求2所述的方法，其特征在于，所述确定各单体压板的边缘直线，包括：

对各单体压板的单体掩码图像进行边缘提取，得到各单体压板的边缘信息；

基于各所述边缘信息，确定各单体压板的边缘轮廓；

对各所述边缘轮廓进行分段，生成对应的轮廓线段；

对所述轮廓线段进行聚类，得到对应各单体压板的边缘直线。

5.根据权利要求4所述的方法，其特征在于，所述对各单体压板的单体掩码图像进行边缘提取，得到各单体压板的边缘信息，包括：

根据所述单体掩码图像，确定单体掩码图像中各像素点的掩码像素值以及坐标位置；

基于各所述掩码像素值，确定对应所述单体压板的目标像素点；

根据所述目标像素点的坐标位置，得到所述单体压板的边缘信息。

6.根据权利要求1所述的方法，其特征在于，所述基于各所述边缘直线，确定所述压板原始图像中各单体压板的压板状态，包括：

根据各单体压板的边缘直线，确定各单体压板的边缘直线夹角；

当所述边缘直线夹角小于第一预设阈值时，确定所述单体压板的压板状态为闭合状态；

当所述边缘直线夹角大于或等于所述第一预设阈值且小于第二预设阈值时,确定所述单体压板的压板状态为半闭合状态；

当所述边缘直线夹角大于或等于所述第二预设阈值时，确定所述单体压板的压板状态为开启状态。

7.根据权利要求1所述的方法，其特征在于，所述生成对应的掩码图像是通过预先训练完成的掩码生成模型生成的，所述掩码生成模型的训练方式，包括：

获取训练图像集，所述训练图像集中包括多张训练图像，各所述训练图像中包括至少一个单体压板；

对各所述训练图像中的压板进行标注，标注压板的轮廓区域以及标签，得到标注后的训练图像集；

将所述标注后的训练图像集输入预先构建的初始掩码生成模型中，通过所述初始掩码生成模型按照标注的标签，对各所述训练图像中的压板进行分割，生成对应的输出掩码图像；

基于所述输出掩码图像以及所述训练图像集，确定所述初始掩码生成模型的模型损失；

根据所述模型损失，对所述初始掩码生成模型进行迭代训练，得到训练完成的掩码模型。

8.一种压板状态确定装置，其特征在于，所述装置包括：

图像采集模块，采集压板原始图像，所述压板原始图像中包括至少一个单体压板；

掩码生成模块，基于所述压板原始图像，生成对应的掩码图像；

掩码图像分析模块，根据所述掩码图像，确定所述压板原始图像中各单体压板的边缘直线；

压板状态分析模块，基于各所述边缘直线，确定所述压板原始图像中各单体压板的压板状态。