基于文本检测模型检测文本框对图像进行旋转校正

  1. 分别对图像旋转0度、22.5度、45度、67.5度,90度,并保存这5张图片。
  2. 使用(轻量)文本检测模型对这5张图片进行文本检测。
  3. 然后挑选长文本框最多、宽高比最大的角度,作为大致的旋转角度判断,得到角度 θ 1 ∈ ( 0 , 22.5 , 45 , 67.5 , 90 ) \theta1 \in (0, 22.5, 45, 67.5, 90) θ1(0,22.5,45,67.5,90)
  4. 使用上一步得到的文本框,统计最大的几个文本框长边倾斜角度得到角度 θ 2 \theta2 θ2
  5. 通过 θ 1 \theta1 θ1 θ 2 \theta2 θ2来对原图进行旋转。

左侧原图,右侧校正后图像,四周使用0填充,校正后图像检测文本框画在上边。
在这里插入图片描述

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/sdlypyzq/article/details/109260598