路面病害识别方法、模型训练方法、电子设备及介质技术

本申请涉及路面养护的领域，尤其是涉及一种路面病害识别方法、模型训练方法、电子设备及介质，该识别方法包括获取路面图像，然后将路面图像输入病害识别模型，得到病害类型，其中病害识别模型是基于第二模型对第一模型进行迁移学习得到的，第一模型是基于训练样本集对初始网络模型训练得到的，训练样本集中包括多个初始图像，每个初始图像为对路面病害所拍摄的图像，且每个初始图像关联一个病害类型标签，第二模型是基于光影训练样本集对初始模型训练得到的，光影训练样本集包括多个在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的路面图像。本申请能够提高对在光线较差的环境下采集的路面图像中路面病害识别的精准度。别的精准度。别的精准度。

【技术实现步骤摘要】
路面病害识别方法、模型训练方法、电子设备及介质


[0001]本申请涉及路面养护的领域，尤其是涉及一种路面病害识别方法、模型训练方法、电子设备及介质。

技术介绍

[0002]道路的路面病害一般包括龟裂、断裂、坑槽等，属于路面养护中的常见状态。道路病害的存在会给交通运输带来较大的安全隐患，因而对道路病害进行精确的检测以便进行及时的修复是至关重要的。
[0003]相关技术中，依靠检测车进行路面图像数据采集，并基于深度学习的路面病害检测模型对收集到的图像数据进行识别检测，检测完成之后系统自动生成路面病害识别报告，包含裂缝类型、破损程度及位置信息的描述。
[0004]相关技术中的检测方案，对在光线较好的白天采集的路面图像中路面病害类型的识别准确度尚可， 但是对于在光线较差的环境下采集的路面图像中路面病害类型的识别准确度较低，如在雨雾天气、傍晚时段采集的路面图像。
[0005]因此，如何提高对在光线较差的环境下采集的路面图像中路面病害识别的精准度，是一个亟需解决的问题。

技术实现思路

[0006]为了提高对在光线较差的环境下采集的路面图像中路面病害识别的精准度，本申请提供一种病害识别方法、模型训练方法、电子设备及介质。
[0007]第一方面，本申请提供一种路面病害识别方法，采用如下的技术方案：一种路面病害识别方法，包括：获取路面图像；将所述路面图像输入病害识别模型，得到病害类型，所述病害识别模型是基于第二模型对第一模型进行迁移学习得到的，所述第一模型是基于训练样本集对初始网络模型训练得到的，所述训练样本集中包括多个初始图像，每个所述初始图像为对路面病害所拍摄的图像，且每个所述初始图像关联一个病害类型标签，所述第二模型是基于光影训练样本集对初始模型训练得到的，所述光影训练样本集包括多个光影图像，所述光影图像为在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的路面图像。
[0008]通过采用上述技术方案，由于病害识别模型是基于第二模型对第一模型进行迁移学习得到的，而第一模型是基于训练样本集对初始网络模型训练得到，第二模型是基于包括多个在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的图像的光影训练样本集得到的，因此，迁移学习后得到的病害识别模型在能够识别路面病害的基础上，提升了对光线不足以及特殊天气下场景的图像的识别能力，而在获取路面图像之后，通过病害识别模型进行病害识别，得到的病害类型是较为准确的。
[0009]第二方面，本申请提供一种模型训练方法，采用如下的技术方案：
一种模型训练方法，包括：获取训练样本集，所述训练样本集中包括多个初始图像，每个所述初始图像为对路面病害所拍摄的图像，且每个所述初始图像关联一个病害类型标签；基于所述训练样本集对初始网络模型进行训练，得到第一模型；基于第二模型对所述第一模型进行迁移学习，得到病害识别模型，所述第二模型是基于在光影训练样本集训练得到的，所述光影训练样本集包括多个光影图像，所述光影图像为在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的路面图像。
[0010]通过采用上述技术方案，由于病害识别模型是基于第二模型对第一模型进行迁移学习得到的，而第一模型是基于训练样本集对初始网络模型训练得到，第二模型是基于包括多个在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的图像的光影训练样本集得到的，因此，迁移学习后得到的病害识别模型在能够识别路面病害的基础上，提升了对光线不足以及特殊天气下场景的图像的识别能力。
[0011]在一种可能实现的方式中，所述基于所述训练样本集对初始网络模型进行训练，得到第一模型，包括：将所述训练样本集进行特征增强预处理，得到每个所述初始图像预处理后对应的训练图像，所述预处理包括多尺度特征提取、注意力计算处理以及膨胀卷积处理；基于每个所述训练图像对初始网络模型进行训练，得到第一模型。
[0012]在一种可能实现的方式中，所述将所述训练样本集进行特征增强预处理，得到每个所述初始图像预处理后对应的训练图像，包括：将每个所述初始图像进行多尺度特征提取，得到每个所述初始图像对应的全局特征，所述全局特征包括所述初始图像在多个图像尺度下各自对应的底层特征，所述底层特征至少包括颜色特征、纹理特征、边缘特征以及形状特征；对每个所述初始图像对应的全局特征进行注意力计算，得到每个所述初始图像在注意力计算后的全局特征；对每个所述初始图像在注意力计算后的全局特征进行膨胀卷积，得到每个所述初始图像对应的训练图像。
[0013]在一种可能实现的方式中，所述对每个所述初始图像对应的全局特征进行注意力计算，得到每个所述初始图像在注意力计算后的全局特征，包括：对每个所述初始图像在每个图像尺度下对应的底层特征分配对应的权重参数；对每个所述初始图像的各个所述底层特征以及各个所述底层特征各自对应的权重参数进行加权平均，得到每个所述初始图像在注意力计算后的全局特征。
[0014]在一种可能实现的方式中，所述将所述训练样本集进行特征增强预处理，得到每个所述初始图像预处理后对应的训练图像，包括：基于GAN的生成器对所述训练样本集进行预处理，得到每个所述初始图像预处理后对应的训练图像。
[0015]在一种可能实现的方式中，所述基于第二模型对所述第一模型进行迁移学习，得到病害识别模型，包括：基于所述第一模型的第一特征提取层和所述第二模型的第二特征提取层进行融合，得到融合后的特征提取层，所述第一模型包括第一特征提取层，所述第二模型包括第二
特征提取层；基于所述融合后的特征提取层以及所述第一模型得到病害识别模型。
[0016]第三方面，本申请提供一种路面病害识别装置，采用如下的技术方案：一种路面病害识别装置，包括：路面图像获取模块，用于获取路面图像；病害识别模块，用于将所述路面图像输入病害识别模型，得到病害类型。
[0017]第四方面，本申请提供一种模型训练装置，采用如下的技术方案：一种模型训练装置，包括：训练样本集获取模块，用于获取训练样本集，所述训练样本集中包括多个初始图像，每个所述初始图像为对路面病害所拍摄的图像，且每个所述初始图像关联一个病害类型标签；训练模块，用于基于所述训练样本集对初始网络模型进行训练，得到第一模型；迁移学习模块，用于基于第二模型对所述第一模型进行迁移学习，得到病害识别模型，所述第二模型是基于在光影训练样本集训练得到的，所述光影训练样本集包括多个光影图像，所述光影图像为在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的路面图像。
[0018]在一种可能实现的方式中，训练模块在基于所述训练样本集对初始网络模型进行训练，得到第一模型时，具体用于：将所述训练样本集进行特征增强预处理，得到每个所述初始图像预处理后对应的训练图像，所述预处理包括多尺度特征提取、注意力计算处理以及膨胀卷积处理；基于每个所述训练图像对初始网络模型进行训练，得到第一模型。
[0019]在一种可能实现的方式中，训练模块在将所述训练样本集进行特征增强预处理，得到本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种路面病害识别方法，其特征在于，包括：获取路面图像；将所述路面图像输入病害识别模型，得到病害类型，所述病害识别模型是基于第二模型对第一模型进行迁移学习得到的，所述第一模型是基于训练样本集对初始网络模型训练得到的，所述训练样本集中包括多个初始图像，每个所述初始图像为对路面病害所拍摄的图像，且每个所述初始图像关联一个病害类型标签，所述第二模型是基于光影训练样本集对初始模型训练得到的，所述光影训练样本集包括多个光影图像，所述光影图像为在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的路面图像。2.一种模型训练方法，其特征在于，包括：获取训练样本集，所述训练样本集中包括多个初始图像，每个所述初始图像为对路面病害所拍摄的图像，且每个所述初始图像关联一个病害类型标签；基于所述训练样本集对初始网络模型进行训练，得到第一模型；基于第二模型对所述第一模型进行迁移学习，得到病害识别模型，所述第二模型是基于在光影训练样本集训练得到的，所述光影训练样本集包括多个光影图像，所述光影图像为在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的路面图像。3.根据权利要求2所述的一种模型训练方法，其特征在于，所述基于所述训练样本集对初始网络模型进行训练，得到第一模型，包括：将所述训练样本集进行特征增强预处理，得到每个所述初始图像预处理后对应的训练图像，所述预处理包括多尺度特征提取、注意力计算处理以及膨胀卷积处理；基于每个所述训练图像对初始网络模型进行训练，得到第一模型。4.根据权利要求3所述的一种模型训练方法，其特征在于，所述将所述训练样本集进行特征增强预处理，得到每个所述初始图像预处理后对应的训练图像，包括：将每个所述初始图像进行多尺度特征提取，得到每个所述初始图像对应的全局特征，所述全局特征包括所述初始图像在多个图像尺度下各自对应的底层特征，所述底层特征至少包括颜色特征、纹理特征、边缘特征以及形状特征；对每个所述初始图像对应的全局特征进行注意力计算，得到每个所述初始图像在注...

【专利技术属性】
技术研发人员：封磊，徐晓龙，俞自飞，冯芳逸，杨亮亮，杨培志，王琳珂，
申请(专利权)人：安徽省路桥试验检测有限公司，
类型：发明
国别省市：