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

(19)中华人民共和国国家知识产权局(12)发明专利申请(10)申请公布号CN113850238A(43)申请公布日2021.12.28(21)申请号202111433042.0(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)权利要求书2页说明书13页附图8页(54)发明名称文档检测方法、装置、电子设备及存储介质(57)摘要本公开提供一种文档检测方法、装置、电子设备及存储介质，所述方法包括：获取待检测图像和所述待检测图像的边缘特征图；将所述待检测图像和所述边缘特征图输入至预先训练的角点检测模型，以获取所述待检测图像的角点预测图；根据所述角点预测图中每个特征点的预测得分，从所述角点预测图中确定目标角点；根据所述目标角点的坐标值，从所述待检测图像中裁剪出文档图像，由此，通过结合待检测图像的边缘特征图来获取角点预测图，进而根据从角点预测图中确定的目标角点，从待检测图像中裁剪出文档图像，能够提高图像中文档检测的精度，实现对图像中的文档进行准确地拆分。CN113850238ACN113850238A权利要求书1/2页1.一种文档检测方法，所述方法包括：获取待检测图像和所述待检测图像的边缘特征图；将所述待检测图像和所述边缘特征图输入至预先训练的角点检测模型，以获取所述待检测图像的角点预测图；根据所述角点预测图中每个特征点的预测得分，从所述角点预测图中确定目标角点；根据所述目标角点的坐标值，从所述待检测图像中裁剪出文档图像。2.根据权利要求1所述的文档检测方法，其中，所述角点预测图包括左上角点预测图、左下角点预测图、右上角点预测图和右下角点预测图中的至少一个，并且其中，所述根据所述角点预测图中每个特征点的预测得分，从所述角点预测图中确定目标角点，包括：将所述角点预测图中预测得分大于预设得分阈值的特征点，确定为所述目标角点。3.根据权利要求2所述的方法，其中，所述目标角点包括目标左上角点、目标左下角点、目标右上角点和目标右下角点中的至少一个，并且其中，所述将所述角点预测图中预测得分大于预设得分阈值的特征点，确定为所述目标角点，包括：将左上角点预测图中预测得分大于所述预设得分阈值的特征点确定为目标左上角点；将左下角点预测图中预测得分大于所述预设得分阈值的特征点确定为目标左下角点；将右上角点预测图中预测得分大于所述预设得分阈值的特征点确定为目标右上角点；将右下角点预测图中预测得分大于所述预设得分阈值的特征点确定为目标右下角点。4.根据权利要求2或3所述的文档检测方法，其中，所述将所述角点预测图中预测得分大于预设得分阈值的特征点，确定为所述目标角点，包括：根据所述角点预测图中每个特征点的预测得分，将预测得分大于所述预设得分阈值的特征点作为候选角点；在所述候选角点为一个的情况下，将所述候选角点确定为所述目标角点；在所述候选角点为至少两个的情况下，将该至少两个候选角点进行合并处理，并将合并后的候选角点作为所述目标角点。5.根据权利要求4所述的文档检测方法，其中，所述将该至少两个候选角点进行合并处理，并将合并后的候选角点作为所述目标角点，包括：根据所述至少两个候选角点中每个候选角点的坐标值，确定所述至少两个候选角点的平均坐标值；将所述平均坐标值对应的坐标点确定为所述目标角点。6.根据权利要求4所述的文档检测方法，其中，在所述候选角点为至少两个的情况下，所述方法还包括：确定至少两个候选角点中第一候选角点与其他候选角点的距离，其中，所述第一候选角点为所述至少两个候选角点中任一候选角点；响应于确定所述第一候选角点与其他候选角点的距离中的最小距离大于预设距离阈值，删除所述第一候选角点。7.根据权利要求1‑3任一项所述的文档检测方法，其中，所述角点检测模型通过如下步骤训练得到：2CN113850238A权利要求书2/2页获取训练样本集，所述训练样本集包括样本图像、与所述样本图像对应的边缘特征图，以及所述样本图像中文档的角点标注数据；将所述样本图像以及与所述样本图像对应的边缘特征图输入待训练模型进行通道融合，得到多维特征数组；对所述多维特征数组进行特征提取，获取与所述样本图像对应的输出特征图；对所述输出特征图进行卷积处理，得到角点预测图；根据所述角点预测图中特征点的预测结果与所述角点标注数据之间的差异，更新所述待训练模型的网络参数，直至所述待训练模型的损失函数值小于或等于预设值，以得到所述角点检测模型。8.一种文档检测装