基于文本检测模型检测文本框对图像进行旋转校正
- 分别对图像旋转0度、22.5度、45度、67.5度,90度,并保存这5张图片。
- 使用(轻量)文本检测模型对这5张图片进行文本检测。
- 然后挑选长文本框最多、宽高比最大的角度,作为大致的旋转角度判断,得到角度 θ 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)。
- 使用上一步得到的文本框,统计最大的几个文本框长边倾斜角度得到角度 θ 2 \theta2 θ2。
- 通过 θ 1 \theta1 θ1和 θ 2 \theta2 θ2来对原图进行旋转。
左侧原图,右侧校正后图像,四周使用0填充,校正后图像检测文本框画在上边。