[发明专利]基于迁移学习技术的室外天气图像分类方法

权利要求书

1.基于迁移学习技术的室外天气图像分类方法，其特征在于包括以下步骤：

1)、建立深度网络模型；

A、在不同的地点布置相机，以固定时间间隔拍取照片图像，然后将获取的图像上传到服务器存储；

B、将获取的图像按照雾、雨、晴、小雪、中雪、大雪分成六类天气候选数据集；

C、六类天气候选数据集的预处理；具体的预处理过程如下所述：首先，从步骤B得到的六类天气候选数据集中分别选取一定数量的图片，然后对选取的图片随机翻转处理一次，扩充一倍的数据量，再利用双线性插值法将原图尺寸调整为适合ResNet的尺寸，然后把调整后的图片分为两部分，一部分为训练集，另一部分为测试集，最后把所有图像数据转换为tfrecord格式文件；

D、将ResNet_V1_50模型在ImageNet竞赛数据集上进行预训练得到ResNet_V1_50预训练模型；

E、将预训练模型进行模型结构的迁移处理，模型结构迁移处理的过程如下所述：将ResNet_V1_50预训练模型的输出层修改为两个分支，每个分支包括一个由3个神经元组成的全连接层，模型结构迁移处理后得到的模型成为第一模型，第一模型的特征提取层结构与ResNet_V1_50预训练模型的特征提取层结构保持一致；

F、对第一模型进行模型参数迁移处理，模型参数迁移处理的过程如下所述：将第一模型的除最后的全连接层参数外的所有参数作为第一模型的初始化参数，第一模型的全连接层参数采用随机初始化，模型参数迁移处理后得到第二模型；

G、对第二模型进行训练；模型训练的具体过程如下所述：采用Adam算法作为训练模型的优化算法，Adam算法的初始学习率设置为0.0001，设置每次迭代的批处理图片数为24，将步骤C得到的tfrecord格式文件的训练集和测试集图像数据导入第二模型中进行训练后得到最终的checkpoint格式的深度网络模型；

2)、将训练好的checkpoint格式的深度网络模型转换为pb格式的深度网络模型；

3)、将待判别的图像输入pb格式的深度网络模型即可得到图像的天气类别。

2.如权利要求1所述的基于迁移学习技术的室外天气图像分类方法，其特征在于：在步骤C中，从步骤B得到的六类天气候选数据集中分别选取的图片数量如下所示：选取雾天、雨天、晴天图像各200张，选取小雪、中雪、大雪天气图像各60张。

3.如权利要求2所述的基于迁移学习技术的室外天气图像分类方法，其特征在于：在步骤C中，利用双线性插值法将原图尺寸调整为224*224。

4.如权利要求1所述的基于迁移学习技术的室外天气图像分类方法，其特征在于：在步骤C中，把调整后的图片分为两部分，训练集的图片数量占比为85％，测试集的图片数量占比为15％。