结合RetinaNet和透视变换的仓库钢铁长材图像自动矫正方法

本发明涉及一种结合RetinaNet和透视变换的仓库钢铁长材图像自动矫正方法，属于机器视觉技术领域。该方法包括图像采集及标注、目标检测模型的构建、平均中点坐标的计算、中点坐标集的划分、矫正坐标的计算、透视变换六大步骤。本发明通过RetinaNet模型对仓库的钢材垛位坐标进行检测识别，提取钢材垛位坐标，计算图像每个钢材垛位的空间关系，得到最佳矫正坐标点，使用矫正坐标点进行透视变换得到矫正图像。本发明方法解决了采集图像受光线、钢材的摆放、拍照位置等影响导致的图像倾斜问题，能够满足复杂场景下各类多样倾斜图像的自动矫正。