Opencv canny边缘检测后的机器学习用于检测四边形？

基本上，我将使用Canny和hough进行文档提取。在Canny边缘检测之后，我得到了很多从文档中检测到的边缘

Canny边缘检测后的结果：

但我只需要文件的边缘，而不仅仅是一个矩形。我使用机器学习搜索对象定位，但它们都以矩形边界框的形式接受输入，而我需要一个四边形边界框。是否有我可以训练和提取文档边缘的机器学习

机器学习边缘检测后的预期输出：

---

在边缘检测之前，只应用强水平模糊，然后垂直重复。

您的问题有了一个良好的开端-但您需要补充：您具体尝试了什么（显示代码！），以及您认为可能导致与预期行为之间差距的更多信息。我现在，我制作了每幅图像的遮罩，这些遮罩是黑白图像，边缘是白色的。我可以在什么机器学习模型上训练这些图像及其遮罩？我假设你的方法是将这两个步骤平行应用于原始图像，然后将它们组合起来？如果水平边缘检测和垂直边缘检测是不相交的，那么这将起作用。是你的经验，他们可以产生和组合干净？