技术特征:
1.一种图像矫正方法,其特征在于,所述方法包括:获取色卡数据集,所述色卡数据集中包括多张角度不同的色卡样本图像以及每张所述色卡样本图像的偏移角度;将所述色卡样本图像作为角度预测模型的输入,将所述色卡样本图像对应的偏移角度作为所述角度预测模型的输出,对所述角度预测模型进行训练,得到训练好的角度预测模型;将待矫正色卡图像输入所述训练好的角度预测模型中,得到所述待矫正色卡图像的偏移角度;根据所述待矫正色卡图像的偏移角度,对所述待矫正色卡图像进行旋转,实现所述待矫正色卡图像的矫正。2.根据权利要求1所述的方法,其特征在于,所述将所述色卡样本图像作为角度预测模型的输入,将所述色卡样本图像对应的偏移角度作为所述角度预测模型的输出,对所述角度预测模型进行训练,得到训练好的角度预测模型,包括:将所述色卡数据集按照预设比例分为训练数据集、验证数据集、测试数据集;将训练数据集中色卡样本图像作为角度预测模型的输入,将所述训练数据集中色卡样本图像对应的偏移角度作为所述角度预测模型的输出,对所述角度预测模型进行训练;根据所述验证数据集以及所述测试数据集对所述角度预测模型的超参数进行调整,得到训练好的角度预测模型。3.根据权利要求2所述的方法,其特征在于,所述根据所述验证数据集以及所述测试数据集对所述角度预测模型的超参数进行调整,得到训练好的角度预测模型,包括:将验证数据集中色卡样本图像作为角度预测模型的输入,得到所述角度预测模型输出的所述验证数据集中色卡样本图像对应的第一偏移角度预测值;根据所述第一偏移角度预测值与所述验证数据集中色卡样本图像对应的偏移角度之间的第一误差,对所述角度预测模型的超参数进行更新,得到更新后的角度预测模型;将测试数据集中色卡样本图像作为所述更新后的角度预测模型的输入,得到所述更新后的角度预测模型输出的所述测试数据集中色卡样本图像对应的第二偏移角度预测值;若根据所述第二偏移角度预测值与所述测试数据集中色卡样本图像对应的偏移角度之间的第二误差,确定所述更新后的角度预测模型的准确率符合预设条件,则确定所述更新后的角度预测模型为训练好的角度预测模型。4.根据权利要求3所述的方法,其特征在于,所述若根据所述第二偏移角度预测值与所述测试数据集中色卡样本图像对应的偏移角度之间的第二误差,确定所述角度预测模型的准确率符合预设条件,则确定所述角度预测模型训练完成,得到训练好的角度预测模型之前,所述方法还包括:计算所述第二偏移角度预测值与所述测试数据集中色卡样本图像对应的偏移角度之间的均方根误差,作为所述数据集中色卡样本图像对应的第二误差。5.根据权利要求2所述的方法,其特征在于,所述角度预测模型的超参数至少包括下列一项或多项:所述角度预测模型的优化器、所述角度预测模型的迭代次数、所述角度预测模型的学习率。
6.根据权利要求3所述的方法,其特征在于,所述根据所述第二偏移角度预测值与所述测试数据集中色卡样本图像对应的偏移角度之间的第二误差,确定所述更新后的角度预测模型的准确率符合预设条件,包括:若所述第二偏移角度预测值与所述色卡样本图像对应的偏移角度之间的第二误差小于第一预设阈值的色卡样本图像数量占所述测试数据集中色卡样本图像数量的比值大于第二预设阈值,则确定所述角度预测模型的准确率符合预设条件。7.一种图像矫正装置,其特征在于,包括:获取模块,用于获取色卡数据集,所述色卡数据集中包括多张角度不同的色卡样本图像以及每张色卡样本图像的偏移角度;训练模块,用于将所述色卡样本图像作为角度预测模型的输入,将所述色卡样本图像对应的偏移角度作为所述角度预测模型的输出,对所述角度预测模型进行训练,得到训练好的角度预测模型;输入模块,用于将待矫正色卡图像输入所述训练好的角度预测模型中,得到所述待矫正色卡图像的偏移角度;矫正模块,用于根据所述待矫正色卡图像的偏移角度,对所述待矫正色卡图像进行旋转,实现所述待矫正色卡图像的矫正。8.根据权利要求7所述的装置,其特征在于,训练模块包括划分单元、训练单元、调整单元;划分单元,用于将所述色卡数据集按照预设比例分为训练数据集、验证数据集、测试数据集;训练单元,用于将训练数据集中色卡样本图像作为角度预测模型的输入,将所述训练数据集中色卡样本图像对应的偏移角度作为所述角度预测模型的输出,对所述角度预测模型进行训练;调整单元,用于根据所述验证数据集以及所述测试数据集对所述角度预测模型的超参数进行调整,得到训练好的角度预测模型。9.一种计算机设备,包括存储器和处理器,所述存储器存储有计算机程序,其特征在于,所述处理器执行所述计算机程序时实现权利要求1至6中任一项所述方法的步骤。10.一种计算机可读存储介质,其上存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现权利要求1至6中任一项所述的方法的步骤。
技术总结
本公开涉及一种图像矫正方法、装置、设备及计算机可读存储介质,该方法包括:获取色卡数据集,所述色卡数据集中包括多张角度不同的色卡样本图像以及每张所述色卡样本图像的偏移角度;将所述色卡样本图像作为角度预测模型的输入,将所述色卡样本图像对应的偏移角度作为所述角度预测模型的输出,对所述角度预测模型进行训练,得到训练好的角度预测模型;将待矫正色卡图像输入所述训练好的角度预测模型中,得到所述待矫正色卡图像的偏移角度;根据所述待矫正色卡图像的偏移角度,对所述待矫正色卡图像进行旋转,实现所述待矫正色卡图像的矫正。本公开通过实现自动化、高效、准确的图像矫正,提高了使用色卡进行图像测试的效率与准确性。确性。确性。
技术研发人员:侯一泽
受保护的技术使用者:上海闻泰信息技术有限公司
技术研发日:2023.06.29
技术公布日:2023/10/10