多文档检测方法、装置、电子设备及存储介质.pdf

(19)中华人民共和国国家知识产权局(12)发明专利申请(10)申请公布号CN113850239A(43)申请公布日2021.12.28(21)申请号202111433044.X(22)申请日2021.11.29(71)申请人北京世纪好未来教育科技有限公司地址100089北京市海淀区中关村大街32号蓝天和盛大厦1702-03室(72)发明人张子浩(74)专利代理机构北京开阳星知识产权代理有限公司11710代理人王艳斌(51)Int.Cl.G06K9/00(2006.01)G06K9/34(2006.01)G06K9/46(2006.01)G06K9/62(2006.01)权利要求书3页说明书15页附图7页(54)发明名称多文档检测方法、装置、电子设备及存储介质(57)摘要本公开提供一种多文档检测方法、装置、电子设备及存储介质，所述方法包括：获取包含至少一个文档的待检测图像，将待检测图像输入至中心点和边检测模型来获取中心点预测图和边预测图，从中心点预测图中确定出至少一个中心点以及从边预测图中确定出多个边缘线段，将至少一个中心点中的每个中心点与多个边缘线段进行匹配，以确定与每个中心点关联的目标边缘线段，计算与每个中心点关联的目标边缘线段之间的交点，得到与每个中心点关联的目标角点，基于与每个中心点关联的目标角点，采用透视变换进行图像裁剪，得到至少一个文档中的每个文档对应的文档图像，由此，可以实现对图像中包含的至少一个文档进行检测，实现了对图像中的多个文档进行拆分。CN113850239ACN113850239A权利要求书1/3页1.一种多文档检测方法，所述方法包括：获取待检测图像，所述待检测图像中包含至少一个文档；将所述待检测图像输入至预先训练的中心点和边检测模型，以获取所述待检测图像的中心点预测图和边预测图；从所述中心点预测图中确定出至少一个中心点，以及，从所述边预测图中确定出多个边缘线段；将所述至少一个中心点中的每个中心点与所述多个边缘线段进行匹配，以确定与所述每个中心点关联的目标边缘线段；计算与所述每个中心点关联的目标边缘线段之间的交点，得到与所述每个中心点关联的目标角点；基于与所述每个中心点关联的目标角点，采用透视变换对所述待检测图像进行裁剪，得到所述至少一个文档中的每个文档对应的文档图像。2.根据权利要求1所述的多文档检测方法，其中，所述从所述中心点预测图中确定出至少一个中心点，包括：根据所述中心点预测图中每个第一特征点的第一预测得分，获取第一预测得分大于第一得分阈值的第一特征点作为候选中心点；对所述候选中心点之间的距离小于预设的距离阈值的候选中心点进行合并处理，得到所述中心点预测图中的至少一个中心点。3.根据权利要求1所述的多文档检测方法，其中，所述边预测图包括上边缘预测图、右边缘预测图、下边缘预测图和左边缘预测图；相应地，所述从所述边预测图中确定出多个边缘线段，包括：根据所述边预测图中每个第二特征点的第二预测得分，获取第二预测得分大于第二得分阈值的目标第二特征点；根据所述目标第二特征点进行霍夫变换，得到所述边预测图中的多个边缘线段，所述多个边缘线段包括所述上边缘预测图中的至少一个上边缘线段、所述右边缘预测图中的至少一个右边缘线段、所述下边缘预测图中的至少一个下边缘线段和所述左边缘预测图中的至少一个左边缘线段。4.根据权利要求3所述的多文档检测方法，其中，所述将所述至少一个中心点中的每个中心点与所述多个边缘线段进行匹配，以确定与所述每个中心点关联的目标边缘线段，包括：计算所述至少一个中心点中的每个中心点到所述多个边缘线段中的每个边缘线段的距离；根据所述每个中心点到所述至少一个上边缘线段的距离，从所述至少一个上边缘线段中确定出与所述每个中心点距离最近的上边缘线段，作为与所述每个中心点关联的目标上边缘线段；根据所述每个中心点到所述至少一个右边缘线段的距离，从所述至少一个右边缘线段中确定出与所述每个中心点距离最近的右边缘线段，作为与所述每个中心点关联的目标右边缘线段；根据所述每个中心点到所述至少一个下边缘线段的距离，从所述至少一个下边缘线段2CN113850239A权利要求书2/3页中确定出与所述每个中心点距离最近的下边缘线段，作为与所述每个中心点关联的目标下边缘线段；以及根据所述每个中心点到所述至少一个左边缘线段的距离，从所述至少一个左边缘线段中确定出与所述每个中心点距离最近的左边缘线段，作为与所述每个中心点关联的目标左边缘线段。5.根据权利要求4所述的多文档检测方法，其中，所述方法还包括：在与所述每个中心点距离最近的右边缘线段为多个的情况下，根据与所述每个中心点距离最近的右边缘线段相对于所述每个中心点的位置，从多个与所述每个中心点距离最近的右边缘线段中，确定出与所述每个中心点关联的目标右边缘线段。6.根据权利要求4所述的多