路面病害识别方法、模型训练方法、电子设备及介质与流程
未命名
10-14
阅读:102
评论:0
1.本技术涉及路面养护的领域,尤其是涉及一种路面病害识别方法、模型训练方法、电子设备及介质。
背景技术:
2.道路的路面病害一般包括龟裂、断裂、坑槽等,属于路面养护中的常见状态。道路病害的存在会给交通运输带来较大的安全隐患,因而对道路病害进行精确的检测以便进行及时的修复是至关重要的。
3.相关技术中,依靠检测车进行路面图像数据采集,并基于深度学习的路面病害检测模型对收集到的图像数据进行识别检测,检测完成之后系统自动生成路面病害识别报告,包含裂缝类型、破损程度及位置信息的描述。
4.相关技术中的检测方案,对在光线较好的白天采集的路面图像中路面病害类型的识别准确度尚可, 但是对于在光线较差的环境下采集的路面图像中路面病害类型的识别准确度较低,如在雨雾天气、傍晚时段采集的路面图像。
5.因此,如何提高对在光线较差的环境下采集的路面图像中路面病害识别的精准度,是一个亟需解决的问题。
技术实现要素:
6.为了提高对在光线较差的环境下采集的路面图像中路面病害识别的精准度,本技术提供一种病害识别方法、模型训练方法、电子设备及介质。
7.第一方面,本技术提供一种路面病害识别方法,采用如下的技术方案:一种路面病害识别方法,包括:获取路面图像;将所述路面图像输入病害识别模型,得到病害类型,所述病害识别模型是基于第二模型对第一模型进行迁移学习得到的,所述第一模型是基于训练样本集对初始网络模型训练得到的,所述训练样本集中包括多个初始图像,每个所述初始图像为对路面病害所拍摄的图像,且每个所述初始图像关联一个病害类型标签,所述第二模型是基于光影训练样本集对初始模型训练得到的,所述光影训练样本集包括多个光影图像,所述光影图像为在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的路面图像。
8.通过采用上述技术方案,由于病害识别模型是基于第二模型对第一模型进行迁移学习得到的,而第一模型是基于训练样本集对初始网络模型训练得到,第二模型是基于包括多个在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的图像的光影训练样本集得到的,因此,迁移学习后得到的病害识别模型在能够识别路面病害的基础上,提升了对光线不足以及特殊天气下场景的图像的识别能力,而在获取路面图像之后,通过病害识别模型进行病害识别,得到的病害类型是较为准确的。
9.第二方面,本技术提供一种模型训练方法,采用如下的技术方案:
一种模型训练方法,包括:获取训练样本集,所述训练样本集中包括多个初始图像,每个所述初始图像为对路面病害所拍摄的图像,且每个所述初始图像关联一个病害类型标签;基于所述训练样本集对初始网络模型进行训练,得到第一模型;基于第二模型对所述第一模型进行迁移学习,得到病害识别模型,所述第二模型是基于在光影训练样本集训练得到的,所述光影训练样本集包括多个光影图像,所述光影图像为在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的路面图像。
10.通过采用上述技术方案,由于病害识别模型是基于第二模型对第一模型进行迁移学习得到的,而第一模型是基于训练样本集对初始网络模型训练得到,第二模型是基于包括多个在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的图像的光影训练样本集得到的,因此,迁移学习后得到的病害识别模型在能够识别路面病害的基础上,提升了对光线不足以及特殊天气下场景的图像的识别能力。
11.在一种可能实现的方式中,所述基于所述训练样本集对初始网络模型进行训练,得到第一模型,包括:将所述训练样本集进行特征增强预处理,得到每个所述初始图像预处理后对应的训练图像,所述预处理包括多尺度特征提取、注意力计算处理以及膨胀卷积处理;基于每个所述训练图像对初始网络模型进行训练,得到第一模型。
12.在一种可能实现的方式中,所述将所述训练样本集进行特征增强预处理,得到每个所述初始图像预处理后对应的训练图像,包括:将每个所述初始图像进行多尺度特征提取,得到每个所述初始图像对应的全局特征,所述全局特征包括所述初始图像在多个图像尺度下各自对应的底层特征,所述底层特征至少包括颜色特征、纹理特征、边缘特征以及形状特征;对每个所述初始图像对应的全局特征进行注意力计算,得到每个所述初始图像在注意力计算后的全局特征;对每个所述初始图像在注意力计算后的全局特征进行膨胀卷积,得到每个所述初始图像对应的训练图像。
13.在一种可能实现的方式中,所述对每个所述初始图像对应的全局特征进行注意力计算,得到每个所述初始图像在注意力计算后的全局特征,包括:对每个所述初始图像在每个图像尺度下对应的底层特征分配对应的权重参数;对每个所述初始图像的各个所述底层特征以及各个所述底层特征各自对应的权重参数进行加权平均,得到每个所述初始图像在注意力计算后的全局特征。
14.在一种可能实现的方式中,所述将所述训练样本集进行特征增强预处理,得到每个所述初始图像预处理后对应的训练图像,包括:基于gan的生成器对所述训练样本集进行预处理,得到每个所述初始图像预处理后对应的训练图像。
15.在一种可能实现的方式中,所述基于第二模型对所述第一模型进行迁移学习,得到病害识别模型,包括:基于所述第一模型的第一特征提取层和所述第二模型的第二特征提取层进行融合,得到融合后的特征提取层,所述第一模型包括第一特征提取层,所述第二模型包括第二
特征提取层;基于所述融合后的特征提取层以及所述第一模型得到病害识别模型。
16.第三方面,本技术提供一种路面病害识别装置,采用如下的技术方案:一种路面病害识别装置,包括:路面图像获取模块,用于获取路面图像;病害识别模块,用于将所述路面图像输入病害识别模型,得到病害类型。
17.第四方面,本技术提供一种模型训练装置,采用如下的技术方案:一种模型训练装置,包括:训练样本集获取模块,用于获取训练样本集,所述训练样本集中包括多个初始图像,每个所述初始图像为对路面病害所拍摄的图像,且每个所述初始图像关联一个病害类型标签;训练模块,用于基于所述训练样本集对初始网络模型进行训练,得到第一模型;迁移学习模块,用于基于第二模型对所述第一模型进行迁移学习,得到病害识别模型,所述第二模型是基于在光影训练样本集训练得到的,所述光影训练样本集包括多个光影图像,所述光影图像为在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的路面图像。
18.在一种可能实现的方式中,训练模块在基于所述训练样本集对初始网络模型进行训练,得到第一模型时,具体用于:将所述训练样本集进行特征增强预处理,得到每个所述初始图像预处理后对应的训练图像,所述预处理包括多尺度特征提取、注意力计算处理以及膨胀卷积处理;基于每个所述训练图像对初始网络模型进行训练,得到第一模型。
19.在一种可能实现的方式中,训练模块在将所述训练样本集进行特征增强预处理,得到每个所述初始图像预处理后对应的训练图像时,具体用于:将每个所述初始图像进行多尺度特征提取,得到每个所述初始图像对应的全局特征,所述全局特征包括所述初始图像在多个图像尺度下各自对应的底层特征,所述底层特征至少包括颜色特征、纹理特征、边缘特征以及形状特征;对每个所述初始图像对应的全局特征进行注意力计算,得到每个所述初始图像在注意力计算后的全局特征;对每个所述初始图像在注意力计算后的全局特征进行膨胀卷积,得到每个所述初始图像对应的训练图像。
20.在一种可能实现的方式中,训练模块在对每个所述初始图像对应的全局特征进行注意力计算,得到每个所述初始图像在注意力计算后的全局特征时,具体用于:对每个所述初始图像在每个图像尺度下对应的底层特征分配对应的权重参数;对每个所述初始图像的各个所述底层特征以及各个所述底层特征各自对应的权重参数进行加权平均,得到每个所述初始图像在注意力计算后的全局特征。
21.在一种可能实现的方式中,训练模块在将所述训练样本集进行特征增强预处理,得到每个所述初始图像预处理后对应的训练图像时,具体用于:基于gan的生成器对所述训练样本集进行预处理,得到每个所述初始图像预处理后对应的训练图像。
22.在一种可能实现的方式中,迁移学习模块在基于第二模型对所述第一模型进行迁移学习,得到病害识别模型时,具体用于:基于所述第一模型的第一特征提取层和所述第二模型的第二特征提取层进行融合,得到融合后的特征提取层,所述第一模型包括第一特征提取层,所述第二模型包括第二特征提取层;基于所述融合后的特征提取层以及所述第一模型得到病害识别模型。
23.第五方面,本技术提供一种电子设备,采用如下的技术方案:一种电子设备,该电子设备包括:至少一个处理器;存储器;至少一个应用程序,其中所述至少一个应用程序被存储在存储器中并被配置为由至少一个处理器执行,所述至少一个应用程序配置用于:执行上述第一方面所述的一种路面病害识别方法和上述第二方面所述的一种模型训练方法。
24.第六方面,本技术提供一种计算机可读存储介质,采用如下的技术方案:一种计算机可读存储介质,包括:存储有能够被处理器加载并执行上述第一方面所述的一种路面病害识别方法和上述第二方面所述的一种模型训练方法的计算机程序。
25.综上所述,本技术包括以下至少一种有益技术效果:由于病害识别模型是基于第二模型对第一模型进行迁移学习得到的,而第一模型是基于训练样本集对初始网络模型训练得到,第二模型是基于包括多个在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的图像的光影训练样本集得到的,因此,迁移学习后得到的病害识别模型在能够识别路面病害的基础上,提升了对光线不足以及特殊天气下场景的图像的识别能力,而在获取路面图像之后,通过病害识别模型进行病害识别,得到的病害类型是较为准确的。
附图说明
26.图1是本技术实施例中路面病害识别方法的流程示意图;图2是本技术实施例中模型训练方法的流程示意图;图3是本技术实施例中路面病害识别装置的结构示意图;图4是本技术实施例中模型训练装置的结构示意图;图5是本技术实施例电子设备的结构示意图
具体实施方式
以下结合附图1-附图5对本技术作进一步详细说明。
27.本领域技术人员在阅读完本说明书后可以根据需要对本实施例做出没有创造性贡献的修改,但只要在本技术的权利要求范围内都受到专利法的保护。
28.为使本技术实施例的目的、技术方案和优点更加清楚,下面将结合本技术实施例中的附图,对本技术实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例是本技术一部分实施例,而不是全部的实施例。基于本技术中的实施例,本领域普通技术人员在没有作出创造性劳动前提下所获得的所有其他实施例,都属于本技术保护的范围。
29.另外,本文中术语“和/或”,仅仅是一种描述关联对象的关联关系,表示可以存在三种关系,例如,a和/或b,可以表示:单独存在a,同时存在a和b,单独存在b这三种情况。另外,本文中字符“/”,如无特殊说明,一般表示前后关联对象是一种“或”的关系。
30.本技术实施例提供了一种路面病害识别方法,由电子设备执行,参照图1,该方法包括步骤s11和步骤s12,其中:步骤s11、获取路面图像;步骤s12、将路面图像输入病害识别模型,得到病害类型。
31.对于本技术实施例,病害识别模型是基于第二模型对第一模型进行迁移学习得到的,第一模型是基于训练样本集对初始网络模型训练得到的,训练样本集中包括多个初始图像,每个初始图像为对路面病害所拍摄的图像,且每个初始图像关联一个病害类型标签,第二模型是基于光影训练样本集对初始模型训练得到的,光影训练样本集包括多个光影图像,光影图像为在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的路面图像。
32.由于病害识别模型是基于第二模型对第一模型进行迁移学习得到的,而第一模型是基于训练样本集对初始网络模型训练得到,第二模型是基于包括多个在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的图像的光影训练样本集得到的,因此,迁移学习后得到的病害识别模型在能够识别路面病害的基础上,提升了对光线不足以及特殊天气下场景的图像的识别能力,而在获取路面图像之后,通过病害识别模型进行病害识别,得到的病害类型是较为准确的。
33.下面从模型训练的角度阐述病害识别模型训练的方法。
34.本技术实施例提供了一种模型训练方法,用于训练得到上述路面病害识别方法实施例中所应用的病害识别模型,模型训练方法,由电子设备执行,参照图2,该方法包括步骤s21-步骤s23,其中:步骤s21、获取训练样本集;对于本技术实施例,训练样本集包括多个训练样本,训练样本集中包括多个初始图像,每个初始图像为对路面病害所拍摄的图像,且每个初始图像关联一个病害类型标签。每个初始图像及对应的病害类型标签作为一个训练样本,其中,部分训练样本为正训练样本,部分训练样本为负训练样本;正训练样本的病害类型标签为对应的初始图像中实际对应的病害类型,负训练样本的病害类型标签与对应的初始图像中实际对应的病害类型不同。
35.步骤s22、基于训练样本集对初始网络模型进行训练,得到第一模型。
36.步骤s23、基于第二模型对第一模型进行迁移学习,得到病害识别模型,第二模型是基于在光影训练样本集训练得到的,光影训练样本集包括多个光影图像,光影图像为在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的路面图像。
37.对于本技术实施例,部分光影图像是在预设类型的天气环境下所拍摄的图像,部分光影图像是在环境中的光线亮度小于预设值的环境下所拍摄的路面图像,如在清晨以及在傍晚时间段针对路面所拍摄的图像。其中,预设的天气类型可以包括雨天、雾天、阴天以及其他特殊类型的天气,本技术实施例中不进行具体限定。亮度对应的预设值本技术实施例中不进行具体限定,其中傍晚时间段以及清晨时间段可基于实际地区对应的环境选择。
38.由于在特殊的预设类型的天气环境下,以及在光线亮度小于预设值的环境下,光线复杂和/或是强度不足,因此所拍摄的路面图像中的色彩以及轮廓甚至边缘和纹理都与在白天光照单一且充足的环境下拍摄的路面图像有所差别。因此,经过包括多个光影图像训练得到的第二模型相较于第一模型,能够对路面图像中的色彩特征、轮廓特征、边缘特征以及纹理特征有着更优的提取和分辨能力。因此,基于第二模型对第一模型进行迁移学习后得到的病害识别模型在能够识别路面病害的基础上,提升了对光线不足以及特殊天气下场景的图像中色彩特征、轮廓特征、边缘特征以及纹理特征的提取和分辨能力。
39.进一步地,基于训练样本集对初始网络模型进行训练,得到第一模型,具体可以包括:将训练样本集进行特征增强预处理,得到每个初始图像预处理后对应的训练图像,然后基于每个训练图像对初始网络模型进行训练,得到第一模型。
40.具体地,预处理包括多尺度特征提取、注意力计算处理以及膨胀卷积处理,将训练样本集进行特征增强预处理,得到每个初始图像预处理后对应的训练图像,包括:将每个初始图像进行多尺度特征提取,得到每个初始图像对应的全局特征;然后对每个初始图像对应的全局特征进行注意力计算,得到每个初始图像在注意力计算后的全局特征;之后对每个初始图像在注意力计算后的全局特征进行膨胀卷积,得到每个初始图像对应的训练图像。
41.具体地,预设多个参照图像尺度,将每个初始图像转换按照各个参照图像尺度进行转换,得到每个初始图像对应的多个参照尺度下各自对应的参照图像,也即一个初始图像对应多个参照图像。将每个初始图像进行多尺度特征提取,即将每个初始图像对应的各个参照图像进行特征提取,得到各个参照图像各自对应的底层特征,其中底层特征至少包括颜色特征、纹理特征、边缘特征以及形状特征,当然也可以包括其他图像的底层特征,对此,本技术实施例中不进行具体限定。
42.初始图像在不同尺度下进行特征提取时,得到的特征精细程度是不一样的。例如,对一个水果拍摄的图像,要确定图像中是苹果还是桃子,在30*30的图像尺度下进行相关特征的提取就足够完成任务了。但是,要是确定图像中苹果的成熟程度或者确定苹果的具体种类,则需要在60*60的图像尺度上提取相关特征才能够作出准确的判断。路面上的病害类型,有些是面积较大,有些是面积较小的,因此,在图像尺度较小时,提取的相关特征能够对面积较大的病害类型作出较准确的判断,而对面积较小的病害类型的判断准确度较低,甚至可能会忽略掉。同理,在图像尺度较大时,提取的相关特征能够对面积较小的病害类型作出较准确的判断,而对面积较大的病害类型的判断准确度较低。因而,通过对初始图像进行多尺度特征提取,能够增大感受视野。
43.进一步地,对每个初始图像在每个图像尺度下对应的底层特征分配对应的权重参数,其中,各项特征对应的权重参数可以是基于图像尺度预设的。然后,对每个初始图像对应的全局特征进行注意力计算,得到每个初始图像在注意力计算后的全局特征,即对每个初始图像在每个图像尺度下对应的底层特征分配对应的权重参数,然后对每个初始图像的各个底层特征以及各个底层特征各自对应的权重参数进行加权平均,得到每个初始图像在注意力计算后的全局特征。
44.进一步地,对每个初始图像对应的全局特征进行膨胀卷积,进一步将初始图像融合,得到初始图像对应的特征增强后的图像,也就是用于对初始网络模型进行训练的训练
图像。为每个训练图像关联训练图像对应的初始图像所对应的病害类型标签,然后基于每个训练图像以及每个训练图像对应的病害类型标签对初始网络模型进行训练,以得到第一模型。
45.进一步地,将训练样本集进行特征增强预处理,得到每个初始图像预处理后对应的训练图像,还可以采用生成式对抗网络gan的生成器实现:如采用u-net模型的生成器,生成器的下采样步骤即为对初始图像的多尺度特征提取,注意力计算步骤和膨胀卷积即为上采样步骤中的特征融合,得到特征增强图像的过程。
46.进一步地,通过包括多个光影图像的光影训练样本在对初始网络模型进行训练得到第二模型的过程中,也可以对光影训练集中每个光影图像进行特征增强预处理,然后基于预处理后的各个光影图像对初始网络模型进行训练,以得到第二模型。具体地,每个光影图像应包含一个对应的类型标签,其中,类型标签可以是路面病害类型标签,也可以是天气类型标签,或者是其他能够表征光影图像信息的标签,但是,所有光影图像的标签类型均应该是统一的,但是对于具体的标签类型,本技术实施例中不进行具体的限定。
47.当然,也可以直接基于各个光影图像以及各个光影图像各自对应的标签对初始网络模型进行训练以得到第二模型。
48.进一步地,在得到第一模型和第二模型之后,分别提取第一模型对应的用于对输入图像进行特征提取的第一特征提取层,以及第二模型对应的用于对输入图像进行特征提取的第二特征提取层。然后基于第一模型的第一特征提取层和第二模型的第二特征提取层进行融合,得到融合后的特征提取层。
49.具体地,是比较第一特征提取层和第二特征提取层的差异参数,然后将差异参数补充到第一模型中的第一特征提取中进行融合,得到病害识别模型。
50.上述实施例从方法流程的角度介绍一种路面病害识别方法和一种模型训练方法,下述实施例从虚拟模块或者虚拟单元的角度介绍了一种路面病害识别装置和一种模型训练装置,具体详见下述实施例。
51.本技术实施例提供一种路面病害识别装置,如图3所示,该路面病害识别装置具体可以包括路面图像获取模块301和病害识别模块302,其中:路面图像获取模块301,用于获取路面图像;病害识别模块302,用于将路面图像输入病害识别模型,得到病害类型。
52.本技术实施例提供一种模型训练装置,如图4所示,该模型训练装置具体可以包括训练样本集获取模块401、训练模块402以及迁移学习模块403,其中:训练样本集获取模块401,用于获取训练样本集,训练样本集中包括多个初始图像,每个初始图像为对路面病害所拍摄的图像,且每个初始图像关联一个病害类型标签;训练模块402,用于基于训练样本集对初始网络模型进行训练,得到第一模型;迁移学习模块403,用于基于第二模型对第一模型进行迁移学习,得到病害识别模型,第二模型是基于在光影训练样本集训练得到的,光影训练样本集包括多个光影图像,光影图像为在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的路面图像。
53.在一种可能实现的方式中,训练模块402在基于训练样本集对初始网络模型进行训练,得到第一模型时,具体用于:将训练样本集进行特征增强预处理,得到每个初始图像预处理后对应的训练图
像,预处理包括多尺度特征提取、注意力计算处理以及膨胀卷积处理;基于每个训练图像对初始网络模型进行训练,得到第一模型。
54.在一种可能实现的方式中,训练模块402在将训练样本集进行特征增强预处理,得到每个初始图像预处理后对应的训练图像时,具体用于:将每个初始图像进行多尺度特征提取,得到每个初始图像对应的全局特征,全局特征包括初始图像在多个图像尺度下各自对应的底层特征,底层特征至少包括颜色特征、纹理特征、边缘特征以及形状特征;对每个初始图像对应的全局特征进行注意力计算,得到每个初始图像在注意力计算后的全局特征;对每个初始图像在注意力计算后的全局特征进行膨胀卷积,得到每个初始图像对应的训练图像。
55.在一种可能实现的方式中,训练模块402在对每个初始图像对应的全局特征进行注意力计算,得到每个初始图像在注意力计算后的全局特征时,具体用于:对每个初始图像在每个图像尺度下对应的底层特征分配对应的权重参数;对每个初始图像的各个底层特征以及各个底层特征各自对应的权重参数进行加权平均,得到每个初始图像在注意力计算后的全局特征。
56.在一种可能实现的方式中,训练模块402在将训练样本集进行特征增强预处理,得到每个初始图像预处理后对应的训练图像时,具体用于:基于gan的生成器对训练样本集进行预处理,得到每个初始图像预处理后对应的训练图像。
57.在一种可能实现的方式中,迁移学习模块403在基于第二模型对第一模型进行迁移学习,得到病害识别模型时,具体用于:基于第一模型的第一特征提取层和第二模型的第二特征提取层进行融合,得到融合后的特征提取层,第一模型包括第一特征提取层,第二模型包括第二特征提取层;基于融合后的特征提取层以及第一模型得到病害识别模型。
58.本技术实施例中提供了一种电子设备,如图5所示,图5所示的电子设备500包括:处理器501和存储器503。其中,处理器501和存储器503相连,如通过总线502相连。可选地,电子设备500还可以包括收发器504。需要说明的是,实际应用中收发器504不限于一个,该电子设备500的结构并不构成对本技术实施例的限定。
59.处理器501可以是cpu(central processing unit,中央处理器),通用处理器,dsp(digital signal processor,数据信号处理器),asic(application specific integrated circuit,专用集成电路),fpga(field programmable gate array,现场可编程门阵列)或者其他可编程逻辑器件、晶体管逻辑器件、硬件部件或者其任意组合。其可以实现或执行结合本技术公开内容所描述的各种示例性的逻辑方框,模块和电路。处理器501也可以是实现计算功能的组合,例如包含一个或多个微处理器组合,dsp和微处理器的组合等。
60.总线502可包括一通路,在上述组件之间传送信息。总线502可以是pci(peripheral component interconnect,外设部件互连标准)总线或eisa(extended industry standard architecture,扩展工业标准结构)总线等。总线502可以分为地址总
线、数据总线、控制总线等。为便于表示,图5中仅用一条粗线表示,但并不表示仅有一根总线或一种类型的总线。
61.存储器503可以是rom(read only memory,只读存储器)或可存储静态信息和指令的其他类型的静态存储设备,ram(random access memory,随机存取存储器)或者可存储信息和指令的其他类型的动态存储设备,也可以是eeprom(electrically erasable programmable read only memory,电可擦可编程只读存储器)、cd-rom(compact disc read only memory,只读光盘)或其他光盘存储、光碟存储(包括压缩光碟、激光碟、光碟、数字通用光碟、蓝光光碟等)、磁盘存储介质或者其他磁存储设备、或者能够用于携带或存储具有指令或数据结构形式的期望的程序代码并能够由计算机存取的任何其他介质,但不限于此。
62.存储器503用于存储执行本技术方案的应用程序代码,并由处理器501来控制执行。处理器501用于执行存储器503中存储的应用程序代码,以实现前述方法实施例所示的内容。
63.其中,电子设备包括但不限于:移动电话、笔记本电脑、数字广播接收器、pda(个人数字助理)、pad(平板电脑)、pmp(便携式多媒体播放器)、车载终端(例如车载导航终端)等等的移动终端以及诸如数字tv、台式计算机等等的固定终端。还可以为服务器等。图5示出的电子设备仅仅是一个示例,不应对本公开实施例的功能和使用范围带来任何限制。
64.本技术实施例提供了一种计算机可读存储介质,该计算机可读存储介质上存储有计算机程序,当其在计算机上运行时,使得计算机可以执行前述路面病害识别方法实施例和模型训练方法实施例中相应内容。
65.应该理解的是,虽然附图的流程图中的各个步骤按照箭头的指示依次显示,但是这些步骤并不是必然按照箭头指示的顺序依次执行。除非本文中有明确的说明,这些步骤的执行并没有严格的顺序限制,其可以以其他的顺序执行。而且,附图的流程图中的至少一部分步骤可以包括多个子步骤或者多个阶段,这些子步骤或者阶段并不必然是在同一时刻执行完成,而是可以在不同的时刻执行,其执行顺序也不必然是依次进行,而是可以与其他步骤或者其他步骤的子步骤或者阶段的至少一部分轮流或者交替地执行。
66.以上仅是本技术的部分实施方式,应当指出,对于本技术领域的普通技术人员来说,在不脱离本技术原理的前提下,还可以做出若干改进和润饰,这些改进和润饰也应视为本技术的保护范围。
技术特征:
1.一种路面病害识别方法,其特征在于,包括:获取路面图像;将所述路面图像输入病害识别模型,得到病害类型,所述病害识别模型是基于第二模型对第一模型进行迁移学习得到的,所述第一模型是基于训练样本集对初始网络模型训练得到的,所述训练样本集中包括多个初始图像,每个所述初始图像为对路面病害所拍摄的图像,且每个所述初始图像关联一个病害类型标签,所述第二模型是基于光影训练样本集对初始模型训练得到的,所述光影训练样本集包括多个光影图像,所述光影图像为在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的路面图像。2.一种模型训练方法,其特征在于,包括:获取训练样本集,所述训练样本集中包括多个初始图像,每个所述初始图像为对路面病害所拍摄的图像,且每个所述初始图像关联一个病害类型标签;基于所述训练样本集对初始网络模型进行训练,得到第一模型;基于第二模型对所述第一模型进行迁移学习,得到病害识别模型,所述第二模型是基于在光影训练样本集训练得到的,所述光影训练样本集包括多个光影图像,所述光影图像为在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的路面图像。3.根据权利要求2所述的一种模型训练方法,其特征在于,所述基于所述训练样本集对初始网络模型进行训练,得到第一模型,包括:将所述训练样本集进行特征增强预处理,得到每个所述初始图像预处理后对应的训练图像,所述预处理包括多尺度特征提取、注意力计算处理以及膨胀卷积处理;基于每个所述训练图像对初始网络模型进行训练,得到第一模型。4.根据权利要求3所述的一种模型训练方法,其特征在于,所述将所述训练样本集进行特征增强预处理,得到每个所述初始图像预处理后对应的训练图像,包括:将每个所述初始图像进行多尺度特征提取,得到每个所述初始图像对应的全局特征,所述全局特征包括所述初始图像在多个图像尺度下各自对应的底层特征,所述底层特征至少包括颜色特征、纹理特征、边缘特征以及形状特征;对每个所述初始图像对应的全局特征进行注意力计算,得到每个所述初始图像在注意力计算后的全局特征;对每个所述初始图像在注意力计算后的全局特征进行膨胀卷积,得到每个所述初始图像对应的训练图像。5.根据权利要求4所述的一种模型训练方法,其特征在于,所述对每个所述初始图像对应的全局特征进行注意力计算,得到每个所述初始图像在注意力计算后的全局特征,包括:对每个所述初始图像在每个图像尺度下对应的底层特征分配对应的权重参数;对每个所述初始图像的各个所述底层特征以及各个所述底层特征各自对应的权重参数进行加权平均,得到每个所述初始图像在注意力计算后的全局特征。6.根据权利要求3所述的一种模型训练方法,其特征在于,所述将所述训练样本集进行特征增强预处理,得到每个所述初始图像预处理后对应的训练图像,包括:基于gan的生成器对所述训练样本集进行预处理,得到每个所述初始图像预处理后对应的训练图像。7.根据权利要求2-6任一项所述的一种模型训练方法,其特征在于,所述基于第二模型
对所述第一模型进行迁移学习,得到病害识别模型,包括:基于所述第一模型的第一特征提取层和所述第二模型的第二特征提取层进行融合,得到融合后的特征提取层,所述第一模型包括第一特征提取层,所述第二模型包括第二特征提取层;基于所述融合后的特征提取层以及所述第一模型得到病害识别模型。8.一种路面病害识别装置,其特征在于,包括:路面图像获取模块,用于获取路面图像;病害识别模块,用于将所述路面图像输入病害识别模型,得到病害类型。9.一种电子设备,其特征在于,该电子设备包括:至少一个处理器;存储器;至少一个应用程序,其中所述至少一个应用程序被存储在存储器中并被配置为由至少一个处理器执行,所述至少一个应用程序配置用于:执行权利要求1或2-7中任一项所述路面病害识别方法。10.一种计算机可读存储介质,其特征在于,包括:存储有能够被处理器加载并执行如权利要求1或2-7中任一项所述路面病害识别方法的计算机程序。
技术总结
本申请涉及路面养护的领域,尤其是涉及一种路面病害识别方法、模型训练方法、电子设备及介质,该识别方法包括获取路面图像,然后将路面图像输入病害识别模型,得到病害类型,其中病害识别模型是基于第二模型对第一模型进行迁移学习得到的,第一模型是基于训练样本集对初始网络模型训练得到的,训练样本集中包括多个初始图像,每个初始图像为对路面病害所拍摄的图像,且每个初始图像关联一个病害类型标签,第二模型是基于光影训练样本集对初始模型训练得到的,光影训练样本集包括多个在预设类型的天气环境下或在光线强度小于预设值的环境下所拍摄的路面图像。本申请能够提高对在光线较差的环境下采集的路面图像中路面病害识别的精准度。别的精准度。别的精准度。
技术研发人员:封磊 徐晓龙 俞自飞 冯芳逸 杨亮亮 杨培志 王琳珂
受保护的技术使用者:安徽省路桥试验检测有限公司
技术研发日:2023.07.27
技术公布日:2023/10/8
版权声明
本文仅代表作者观点,不代表航空之家立场。
本文系作者授权航家号发表,未经原创作者书面授权,任何单位或个人不得引用、复制、转载、摘编、链接或以其他任何方式复制发表。任何单位或个人在获得书面授权使用航空之家内容时,须注明作者及来源 “航空之家”。如非法使用航空之家的部分或全部内容的,航空之家将依法追究其法律责任。(航空之家官方QQ:2926969996)
飞行汽车 https://www.autovtol.com/
