一种基于UNet算法的面向文本线段检测方法

本发明为一种基于UNet算法的面向文本线段检测方法，具体过程是：获取带线段文本图像数据；对采集的图像进行尺寸归一化处理，利用标注工具labelme对图像进行人工标注，获取线段位置，并扩充为宽度为两个像素点的矩形，获取矩形四点的坐标，再对图像进行二值化处理，生成二值化标签图像，得到目标检测数据集；二值化标签图像中标注原始图像中所有线段所在位置；基于UNet算法构建线段检测网络模型，以目标检测数据集训练线段检测网络模型，以训练好的线段检测网络模型对文本图像中线段进行预测，输出模型预测图；对模型预测图进行后处理，提取文本图像中横向线段、纵向线段、斜向线段的位置，实现在文本图像中线段的检测，提高了文本线段的检测精度。