[发明专利]曲线数据的提取方法、装置、计算机设备、存储介质

权利要求书

1.一种曲线数据的提取方法，其特征在于，所述方法包括：

获取待处理的曲线图像，所述曲线图像中包括坐标系以及位于所述坐标系内的目标曲线，所述坐标系中包括坐标轴和刻度文本标识；

将所述曲线图像输入至预训练的语义分割模型，通过所述语义分割模型确定出所述曲线图像中每个像素点的分类结果，根据所述分类结果对多个所述像素点聚类得到所述坐标轴对应的轴位置以及所述目标曲线对应的曲线位置；

对所述曲线图像进行文本识别处理，确定出所述刻度文本标识的刻度位置以及刻度结果；

根据所述轴位置和所述刻度位置对所述坐标轴和所述刻度文本标识进行匹配，确定所述坐标系；

根据所述曲线位置确定出所述目标曲线在所述坐标系中的相对位置，根据所述坐标系中与所述相对位置对应的目标刻度结果生成与所述目标曲线对应的曲线数据。

2.根据权利要求1所述的方法，其特征在于，所述语义分割模型包括坐标轴分割模型以及曲线分割模型；

所述将所述曲线图像输入至预训练的语义分割模型，通过所述语义分割模型确定出所述曲线图像中每个像素点的分类结果，根据所述分类结果对多个所述像素点聚类得到所述坐标轴对应的轴位置以及所述目标曲线对应的曲线位置，包括：

将所述曲线图像输入至所述坐标轴分割模型，通过所述坐标轴分割模型对所述曲线图像进行坐标轴检测，生成对每个所述像素点的坐标轴分类结果；

通过所述坐标轴分割模型根据所述坐标轴分类结果从所述曲线图像中确定出属于坐标轴类别下的第一像素点，对所述第一像素点进行聚类得到所述坐标轴对应的轴位置；

将所述曲线图像输入至所述曲线分割模型，通过所述曲线分割模型对所述曲线图像进行曲线检测，生成对每个所述像素点的曲线分类结果；

通过所述曲线分割模型根据所述曲线分类结果从所述曲线图像中确定出属于曲线类别下的第二像素点，对所述第二像素点进行聚类得到所述目标曲线对应的曲线位置。

3.根据权利要求2所述的方法，其特征在于，所述目标曲线的曲线类别包括多种；

所述通过所述曲线分割模型根据所述曲线分类结果从所述曲线图像中确定出属于曲线类别下的第二像素点，对所述第二像素点进行聚类得到所述目标曲线对应的曲线位置，包括：

通过所述曲线分割模型根据所述曲线分类结果从所述曲线图像中确定出属于同一曲线类别下的第二像素点，分别对所述同一曲线类别下的第二像素点进行聚类，得到与每种所述曲线类别对应的目标曲线的曲线位置。

4.根据权利要求2所述的方法，其特征在于，所述通过所述坐标轴分割模型根据所述坐标轴分类结果从所述曲线图像中确定出属于坐标轴类别下的第一像素点，对所述第一像素点进行聚类得到所述坐标轴对应的轴位置，包括：

通过所述坐标轴分割模型根据所述坐标轴分类结果从所述曲线图像中确定出属于横轴类别下的第一横轴像素点，以及属于纵轴类别下的第一纵轴像素点；

通过所述坐标轴分割模型对所述第一横轴像素点进行聚类得到与横轴对应的横轴位置，对所述第一纵轴像素点进行聚类得到与纵轴对应的纵轴位置；

通过所述坐标轴分割模型根据所述横轴位置和所述纵轴位置确定出所述坐标系的原点位置。

5.根据权利要求4所述的方法，其特征在于，所述根据所述轴位置和所述刻度位置对所述坐标轴和所述刻度文本标识进行匹配，确定所述坐标系，包括：

根据所述刻度位置和所述横轴位置确定出所述刻度文本标识与所述横轴之间的第一间距，根据所述刻度位置和所述纵轴位置确定出所述刻度文本标识与所述纵轴之间的第二间距；

响应于所述第一间距小于预设间距阈值的比较结果，将所述刻度文本标识和所述横轴匹配，建立所述刻度结果与横轴刻度之间的第一映射关系；或者，

响应于所述第二间距小于所述预设间距阈值的比较结果，将所述刻度文本标识和所述纵轴匹配，建立所述刻度结果与纵轴刻度之间的第二映射关系；

根据所述横轴位置、所述原点位置以及所述纵轴位置确定所述坐标系的坐标系位置，根据所述第一映射关系和所述第二映射关系确定与所述坐标系对应的刻度映射关系，采用所述坐标系位置和所述刻度映射关系确定所述坐标系。