[发明专利]鱼眼图像的矫正模型的微调方法、装置及电子设备

权利要求书

1.一种鱼眼图像的矫正模型的微调方法，其特征在于，包括：

根据相机的内参数、畸变系数构建第一图像样本集；

根据所述第一图像样本集对生成式对抗网络进行训练，得到训练后的生成式对抗网络；

获取预先训练好的用于鱼眼图像的矫正的生成式对抗网络；其中，所述生成式对抗网络包括鱼眼图像判别模型和第一鱼眼图像矫正模型；

根据所述第一鱼眼图像矫正模型构建第二鱼眼图像矫正模型；其中，所述第一鱼眼图像矫正模型为教师模型，所述第二鱼眼图像矫正模型为学生模型；

根据所述第一鱼眼图像矫正模型对所述第二鱼眼图像矫正模型进行知识蒸馏，得到蒸馏后的第二鱼眼图像矫正模型；

其中，所述根据相机的内参数、畸变系数构建第一图像样本集，包括：

根据棋盘标定法对所述相机进行标定，得到所述相机的内参数和畸变系数；

根据所述内参数、所述畸变系数对第一图像进行处理，得到第二图像；其中，所述第二图像为所述第一图像的畸变图像；

根据所述第一图像、所述第二图像构建所述第一图像样本集；

其中，所述根据所述第一图像样本集对所述生成式对抗网络进行训练，得到训练后的生成式对抗网络，包括：

将所述第二图像输入至所述第一鱼眼图像矫正模型中，得到第三图像；

根据所述第一图像、所述第三图像构建第二图像样本集；

根据所述图像样本集对所述鱼眼图像判别模型进行训练，得到训练后的鱼眼图像判别模型；

根据所述第二图像对所述第一鱼眼图像矫正模型进行训练，得到训练后的第一鱼眼图像矫正模型。

2.根据权利要求1所述的鱼眼图像的矫正模型的微调方法，其特征在于，所述根据所述第一鱼眼图像矫正模型构建第二鱼眼图像矫正模型，包括：

对所述第一鱼眼图像矫正模型进行网络参数修剪，得到第三图像矫正模型；

对所述第三图像矫正模型进行知识蒸馏，得到蒸馏后的第三图像矫正模型；

对所述蒸馏后的第三图像矫正模型进行网络参数修剪，得到所述第二鱼眼图像矫正模型。

3.根据权利要求1所述的鱼眼图像的矫正模型的微调方法，其特征在于，所述根据所述第一鱼眼图像矫正模型对所述第二鱼眼图像矫正模型进行知识蒸馏，得到蒸馏后的第二鱼眼图像矫正模型，包括：

获取所述第一鱼眼图像矫正模型输出的第一预测值以及所述第二鱼眼图像矫正模型输出的第二预测值；

根据所述第一预测值、所述第二预测值确定蒸馏后的第二鱼眼图像矫正模型。

4.根据权利要求1所述的鱼眼图像的矫正模型的微调方法，其特征在于，在所述根据所述第一鱼眼图像矫正模型对所述第二鱼眼图像矫正模型进行知识蒸馏，得到蒸馏后的第二鱼眼图像矫正模型之后，还包括：

将蒸馏后的第二鱼眼图像矫正模型部署至终端设备中；

根据所述终端设备中蒸馏后的第二鱼眼图像矫正模型对待矫正图像进行矫正，得到矫正后的图像。