本发明涉及模型训练领域，具体而言，涉及一种落石检测模型训练方法、落石检测方法及相关装置。
背景技术：
落石检测通常都采用雷达或其他传感器进行检测，但通过传感器所采集的信息易受到外界因素干扰，影响判断准确率。
随着近年来计算机视觉的发展，越来越多的人采用计算机视觉的方式对落石进行检测。通过雷达与视频分析技术，对铁路落石进行检测，综合帧间差与雷达信号降低误报率。或者采用机器视觉的方法对隧道内的落石进行检测，采用划分检测区域规避误报。
可以发现，在对落石检测采用计算机视觉的方法时，大多数研究者都采用了视频帧间差作为依据，为了规避其他变化带来的误差，甚至划分了检测区域，但这些方法都不适用与高速道路的落石检测。若使用帧间法进行检测，高速道路上运动的车辆会直接影响结果，且检测区域与车辆运动区域重叠，无法通过划分区域规避误报。
技术实现要素：
本发明的目的包括，例如，提供了一种落石检测模型训练方法、落石检测方法及相关装置，其能够提高落石检测的精确度。
为了实现上述目的，本申请实施例采用的技术方案如下：
第一方面，本申请实施例提供了一种落石检测模型训练方法，所述方法包括：
计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息，其中，所述目标图像的区域标识由图像分割网络输出，所述标准图像的区域标识由人工标注，所述区域标识指示所述目标图像或所述标准图像中的高速道路区域和非高速道路区域，所述第一损失惩罚信息为目标图像中的区域标识与标准图像中的区域标识的差异；
计算所述目标图像中目标落石信息与所述标准图像中标准落石信息的第二损失惩罚项信息，其中，所述第二损失惩罚项信息为目标图像中的目标落石信息与所述标准图像的标准落石信息的差异，所述标准图像的标准落石信息由人工标注；
将所述第一损失惩罚项信息和所述损失第二惩罚项信息分别输入至图像分割网络和落石检测网络中，得到更新后的目标图像的区域标识和更新后的所述目标图像的目标落石信息；
返回执行所述计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息的步骤，直至最新的第一惩罚项信息和最新的第二惩罚项信息满足落石检测条件。
在可选的实施方式中，所述计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息包括：
计算二分类损失函数lseg；
其中，y为样本标签，当标准图像的像素属于高速道路时取值为1，否则为0，y′为模型预测为高速道路类的概率，n为像素点数量。
在可选的实施方式中，所述计算所述目标图像中目标落石信息与所述标准图像中标准落石信息的第二损失惩罚项信息包括：
计算中心坐标误差la,b；
其中，a,b为目标框的中心的横坐标和纵坐标，k*k代表网格数量，m为每个网格所产生的候选框，ai,bi为真实目标框的中心的横坐标和纵坐标，a′i,b′i为目标框的中心的横坐标和纵坐标的预测值，
计算宽高误差lw,h；
其中，w,h为目标框的宽度和高度，wi，hi为目标框的真实宽度和高度，w′i，h′i为目标框宽度和高度的预测值；
计算置信度误差lconf：
其中，
计算分类误差lclass；
p′i(c)为类别为c的真实标签，pi(c)为类别为c的预测值，c为类别为落石的值；
将l＝la,b lw,h lconf lclass作为第二损失函数惩罚项。
在可选的实施方式中，所述图像分割网络的输出作为所述落石检测网络的输入。
在可选的实施方式中，所述图像分割网络由四个卷积层与残差结构组成。
在可选的实施方式中，所述方法还包括：
计算所述落石检测网络的第一平均精度和第二平均精度；
所述计算所述落石检测网络的第一平均精度和第二平均精度的步骤包括：
计算第一平均精度；
其中，p和r分别代表精准度与召回率；
所述精准度
所述召回率
tp表示目标框对应的图像区域的实际值标签为落石，算法预测的类别是落石的目标框个数，fp表示目标框对应的图像区域的实际值标签为非落石，算法预测的类别是落石的目标框个数，fn表示目标框对应的图像区域的实际值标签为落石，算法预测的类别是非落石的目标框个数；
计算第二平均精度；
根据所述第一平均精度和所述第二平均精度对所述落石检测模型进行评价。
第二方面，本申请实施例提供了一种落石检测方法，所述方法包括：
将目标图像输入至图像分割网络；
所述图像分割网络检测所述目标图像中的区域标识，其中，所述区域标识指示所述目标图像的高速道路区域和非高速道路区域；
将携带有区域标识输入的所述目标图像输入至落石检测网络中；
所述落石检测网络对所述目标图像以及所述目标图像对应的标识信息进行检测；
所述落石检测网络输出高速道路区域的落石信息和非高速道路的落石信息。
第三方面，本申请提供了一种落石检测模型训练装置，所述装置包括：
第一计算模块，用于计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息，其中，所述目标图像的区域标识由图像分割网络输出，所述标准图像的区域标识由人工标注，所述区域标识指示所述目标图像或所述标准图像中的高速道路区域和非高速道路区域，所述第一损失惩罚信息为目标图像中的区域标识与标准图像中的区域标识的差异；
第二计算模块，用于计算所述目标图像中目标落石信息与所述标准图像中标准落石信息的第二损失惩罚项信息，其中，所述第二损失惩罚项信息为目标图像中的落石信息与所述标准图像的标准落石信息的差异；
输入模块，用于将所述第一损失惩罚项信息和所述损失第二惩罚项信息分别输入至图像分割网络和落石检测网络中，得到更新后的目标图像和更新后的所述目标图像的目标落石信息；
执行模块，用于返回执行所述计算目标图像和标准图像的第一损失惩罚项信息的步骤，直至最新的第一惩罚项信息和最新的第二惩罚项信息满足落石检测条件。
第四方面，本申请实施例提供了一种落石检测设备，所述设备包括：第一输入模块，用于将目标图像输入至所述的图像分割网络；
第一检测模块，用于所述图像分割网络检测所述目标图像中的区域标识，其中，所述区域标识指示所述目标图像的高速道路区域和非高速道路区域；
第二输入模块，用于将携带有区域标识输入的所述目标图像输入至所述的落石检测网络中；
第二检测模块，用于所述落石检测网络对所述目标图像以及所述目标图像对应的标识信息进行检测；
输出模块，用于所述落石检测网络输出高速道路区域的落石信息和非高速道路的落石信息。
第五方面，本申请实施例提供了一种电子设备，包括存储器和处理器，所述存储器存储有计算机程序，所述处理器执行所述计算机程序时实现所述落石检测模型训练方法的步骤。
第六方面，本申请实施例提供了一种计算机可读存储介质，其上存储有计算机程序，该计算机程序被处理器执行时实现所述落石检测模型训练方法的步骤。
本申请具有以下有益效果：
本申请通过计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息，计算目标图像中目标落石信息与标准图像中标准落石信息的第二损失惩罚项信息；将第一损失惩罚项信息和损失第二惩罚项信息分别输入至图像分割网络和落石检测网络中，得到更新后的目标图像的区域标识和更新后的目标图像的目标落石信息；返回执行计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息的步骤，直至最新的第一惩罚项信息和最新的第二惩罚项信息满足落石检测条件。本申请通过图像分割网络，将落石的注意力放在高速道路上，从而减少高速道路外的误判情况。高速道路的落石检测对于提高山区驾驶安全有着重要作用。
附图说明
为了更清楚地说明本发明实施例的技术方案，下面将对实施例中所需要使用的附图作简单地介绍，应当理解，以下附图仅示出了本发明的某些实施例，因此不应被看作是对范围的限定，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他相关的附图。
图1为本发明实施例提供的电子设备的方框示意图；
图2为本发明实施例提供的一种落石检测模型训练方法的流程示意图之一；
图3为本发明实施例提供图像分割网络示意图；
图4为本发明实施例提供的落石检测网络示意图；
图5为本发明实施例提供的添加图像分割网络的落石检测网络示意图；
图6为本发明实施例提供的一种落石检测模型训练方法的流程示意图之二；
图7为本发明实施例提供的添加图像分割网络前后的特征图对比图；
图8为本发明实施例提供的一种落石检测方法的流程示意图；
图9为本发明实施例提供的基于训练后的模型的检测效果图；
图10为本发明实施例提供的一种落石检测模型训练装置的结构框图；
图11为本发明实施例提供的一种落石检测设备的结构框图。
具体实施方式
为使本发明实施例的目的、技术方案和优点更加清楚，下面将结合本发明实施例中的附图，对本发明实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例是本发明一部分实施例，而不是全部的实施例。通常在此处附图中描述和示出的本发明实施例的组件可以以各种不同的配置来布置和设计。
因此，以下对在附图中提供的本发明的实施例的详细描述并非旨在限制要求保护的本发明的范围，而是仅仅表示本发明的选定实施例。基于本发明中的实施例，本领域普通技术人员在没有作出创造性劳动前提下所获得的所有其他实施例，都属于本发明保护的范围。
应注意到：相似的标号和字母在下面的附图中表示类似项，因此，一旦某一项在一个附图中被定义，则在随后的附图中不需要对其进行进一步定义和解释。
在本发明的描述中，需要说明的是，若出现术语“上”、“下”、“内”、“外”等指示的方位或位置关系为基于附图所示的方位或位置关系，或者是该发明产品使用时惯常摆放的方位或位置关系，仅是为了便于描述本发明和简化描述，而不是指示或暗示所指的装置或元件必须具有特定的方位、以特定的方位构造和操作，因此不能理解为对本发明的限制。
此外，若出现术语“第一”、“第二”等仅用于区分描述，而不能理解为指示或暗示相对重要性。
需要说明的是，在不冲突的情况下，本发明的实施例中的特征可以相互结合。
经过发明人大量研究发现，现有技术中对落石检测采用计算机视觉的方法时，大多数研究者都采用了视频帧间差作为依据，为了规避其他变化带来的误差，甚至划分了检测区域，但这些方法都不适用与高速道路的落石检测。若使用帧间法进行检测，高速道路上运动的车辆会直接影响结果，且检测区域与车辆运动区域重叠，无法通过划分区域规避误报。
有鉴于对上述问题的发现，本实施例提供了一种落石检测模型训练方法，能够通过计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息，其中，所述目标图像的区域标识由图像分割网络输出，所述标准图像的区域标识由人工标注，所述区域标识指示所述目标图像或所述标准图像中的高速道路区域和非高速道路区域，所述第一损失惩罚信息为目标图像中的区域标识与标准图像中的区域标识的差异；计算所述目标图像中目标落石信息与所述标准图像中标准落石信息的第二损失惩罚项信息，其中，所述第二损失惩罚项信息为目标图像中的目标落石信息与所述标准图像的标准落石信息的差异，所述标准图像的标准落石信息由人工标注；将所述第一损失惩罚项信息和所述损失第二惩罚项信息分别输入至图像分割网络和落石检测网络中，得到更新后的目标图像的区域标识和更新后的所述目标图像的目标落石信息；返回执行所述计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息的步骤，直至最新的第一惩罚项信息和最新的第二惩罚项信息满足落石检测条件，基于训练好的网络，对落石进行检测。
本实施例提供一种可以对落石检测模型进行训练的电子设备。在一种可能的实现方式中，所述电子设备可以为用户终端，例如，电子设备可以是，但不限于，服务器、智能手机、个人电脑(personalcomputer，pc)、平板电脑、个人数字助理(personaldigitalassistant，pda)、移动上网设备(mobileinternetdevice，mid)以及图像采集装置等。
请参照图1，图1是本申请实施例提供的电子设备100的结构示意图。所述电子设备100还可包括比图1中所示更多或者更少的组件，或者具有与图1所示不同的配置。图1中所示的各组件可以采用硬件、软件或其组合实现。
所述电子设备100包括落石检测模型训练装置110、存储器120及处理器130。
所述存储器120及处理器130各元件相互之间直接或间接地电性连接，以实现数据的传输或交互。例如，这些元件相互之间可通过一条或多条通讯总线或信号线实现电性连接。所述落石检测模型训练装置110包括至少一个可以软件或固件(firmware)的形式存储于所述存储器120中或固化在所述电子设备100的操作系统(operatingsystem，os)中的软件功能模块。所述处理器130用于执行所述存储器120中存储的可执行模块，例如所述基于落石检测模型训练装置110所包括的软件功能模块及计算机程序等。
其中，所述存储器120可以是，但不限于，随机存取存储器(randomaccessmemory，ram)，只读存储器(readonlymemory，rom)，可编程只读存储器(programmableread-onlymemory，prom)，可擦除只读存储器(erasableprogrammableread-onlymemory，eprom)，电可擦除只读存储器(electricerasableprogrammableread-onlymemory，eeprom)等。其中，存储器120用于存储程序，所述处理器130在接收到执行指令后，执行所述程序。
请参照图2，图2为应该用于图1的电子设备100的一种落石检测模型训练方法的流程图，以下将方法包括各个步骤进行详细阐述。
步骤201：计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息。
其中，目标图像的区域标识由图像分割网络输出，标准图像的区域标识由人工标注，区域标识指示目标图像或标准图像中的高速道路区域和非高速道路区域，第一损失惩罚信息为目标图像中的区域标识与标准图像中的区域标识的差异。
步骤202：计算目标图像中目标落石信息与标准图像中标准落石信息的第二损失惩罚项信息。
其中，第二损失惩罚项信息为目标图像中的目标落石信息与标准图像的标准落石信息的差异，标准图像的标准落石信息由人工标注。
步骤203：将第一损失惩罚项信息和损失第二惩罚项信息分别输入至图像分割网络和落石检测网络中，得到更新后的目标图像的区域标识和更新后的目标图像的目标落石信息。
步骤204：返回执行计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息的步骤，直至最新的第一惩罚项信息和最新的第二惩罚项信息满足落石检测条件。
当最新的第一惩罚项信息和最新的第二惩罚项信息收敛时，则满足落石检测条件，即第一惩罚项信息收敛时，图像分割网络可以对图像的区域准确识别，当更新的第二惩罚项信息收敛时，落石检测网络可以对图像中的落石进行准确检测。
如图3所示，为图像分割网络示意图。
图像分割网络由四个卷积层与残差结构组成。
目标图像通过第一卷积层、第二卷积层、第三卷积层以及第四卷积层，在残差结构层输出图像的结果，即输出携带有区域标识的目标图像。
标准图像的区域标识是通过人为标注的，即人为标注标准图像中的高速道路区域和非高速道路区域。目标图像的区域标识是通过图像分割网络预测得到的，将目标图像输入至图像分割网络，图像分割网络依据目标图像生成预测参数，基于预测参数对目标图像的区域进行标注，得到目标图像的区域标识。
标准图像的标准落石信息是通过人为标注的，即人为标注标准图像中的标准落石信息，目标图像的目标落石信息是通过落石检测网络预测得到的，将目标图像输入至落石检测网络，落石检测网络依据目标图像生成预测参数，基于预测参数对目标图像中的落石进行标注，得到目标图像的目标落石信息。
图4是落石检测网络示意图。落石检测网络采用yolov3网络结构，yolov3由多个dbl与res结构组成，并采用多尺度的方式进行3个尺度的检测。
dbl:代码中的darknetconv2d_bn_leaky，是yolo_v3的基本组件。就是卷积 bn leakyrelu。resn：n代表数字，有res1，res2,…,res8等等，表示这个res_block里含有多少个res_unit。
携带有区域标识的目标图像输入至yolov3网络中，输出目标落石信息。如图5所示，为添加图像分割网络的落石检测网络示意图。
将图像分割网络与yolov3网络结合，图像分割网络的输出作为yolov3网络的输入。
目标图像经过图像分割网络，得到携带有区域标识的目标图像，将目标图像作为输入，输入至yolov3网络中进行落石检测，得到目标图像中高速道路区域中的落石信息和非高速道路区域的落石信息。
计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息具体为：
计算二分类损失函数lseg；
其中，y为样本标签，当标准图像的像素属于高速道路时取值为1，否则为0，y′为模型预测为高速道路类的概率，n为像素点数量。
计算目标图像中目标落石信息与标准图像中标准落石信息的第二损失惩罚项信息具体为：
计算中心坐标误差la,b；
其中，a,b为目标框的中心的横坐标和纵坐标，k*k代表网格数量，m为每个网格所产生的候选框，ai,bi为真实目标框的中心的横坐标和纵坐标，a′i,b′i为目标框的中心的横坐标和纵坐标的预测值，
计算宽高误差lw,h；
其中，w,h为目标框的宽度和高度，wi，hi为目标框的真实宽度和高度，w′i，h′i为目标框宽度和高度的预测值；
计算置信度误差lconf；
其中，
对于一幅图像，一般而言大部分内容是不包含待检测类别的目标物体的，导致没有待检测类别的目标物体的图像计算部分贡献会大于有物体的计算部分，这会导致网络倾向于预测单元格不含有物体。因此要减少没有物体计算部分的贡献权重，比如取值为：0.5。
落石检测网络最终会产生一些特征图，特征图对应的就是一张网格，网格的大小是k*k，每个格子上会产生m个候选框。i是k*k的网格的遍历，j是每个格子上的m个候选框的遍历。一共产生k*k*m个候选框，但是这些候选框中只有与真实目标框的iou大于一定值(阈值)的时候，才会纳入这个la,b的损失函数的计算。阈值的设置通常通过人工经验设置，比如阈值设置为0.2或0.3。
计算分类误差lclass：
p′i(c)为类别为c的真实标签，pi(c)为类别为c的预测值，c为类别为落石的值；
将l＝la,b lw,h lconf lclass作为第二损失函数惩罚项。
在统计学中，一个概率样本的置信区间(confidenceinterval)是对这个样本的某个总体参数的区间估计。置信区间展现的是这个参数的真实值有一定概率落在测量结果的周围的程度。置信区间给出的是被测量参数测量值的可信程度范围，即前面所要求的“一定概率”。这个概率被称为置信水平。
依据第一惩罚项信息对图像分割网络的预测参数进行调整，依据第二惩罚项信息对落石检测网络的预测参数进行调整，直至第一惩罚项信息和第二惩罚项信息最小为止。
本申请通过计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息，计算目标图像中目标落石信息与标准图像中标准落石信息的第二损失惩罚项信息；将第一损失惩罚项信息和损失第二惩罚项信息分别输入至图像分割网络和落石检测网络中，得到更新后的目标图像的区域标识和更新后的目标图像的目标落石信息；返回执行计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息的步骤，直至最新的第一惩罚项信息和最新的第二惩罚项信息满足落石检测条件。本申请通过图像分割网络，将落石的注意力放在高速道路上，从而减少高速道路外的误判情况。高速道路的落石检测对于提高山区驾驶安全有着重要作用。
为了对训练好的模型进行评价，在本申请的另一个实施例中，如图6所示，提供了一种落石检测模型训练方法，具体包括以下步骤：
步骤301：计算落石检测网络的第一平均精度和第二平均精度。
具体为：
其中，p和r分别代表精准度与召回率；
精准度
召回率
tp表示目标框对应的图像区域的实际值标签为落石，算法预测的类别是落石的目标框个数，fp表示目标框对应的图像区域的实际值标签为非落石，算法预测的类别是落石的目标框个数，fn表示目标框对应的图像区域的实际值标签为落石，算法预测的类别是非落石的目标框个数。
计算第二平均精度具体为：
步骤302：根据第一平均精度和第二平均精度对落石检测模型进行评价。
深度学习测试模型效果的时,经常会使用一些特定的评价指标。根据数据集的分布情况以及使用侧重的效果可以选择不同的指标进行评价。一般常用的评价指标有第一平均精度、第二平均精度、map、roc/auc。
第一平均精度衡量的是训练出的模型在每个类别上的好坏。
p，准确率/查准率。r，召回率/查全率。这两个指标分别以两个角度衡量分类系统的准确率。目标检测中计算查准率和召回率之前需要需要计算检测框的iou值,并根iou值判断是否预测正确。
p-r曲线(pr曲线),pr曲线的横轴r也就是tpr，反映了分类器对正例的覆盖能力。而纵轴p的分母是识别为正例的数目，而不是实际正例数目。r反映了分类器预测正例的准确程度。那么，p-r曲线反映了分类器对正例的识别准确程度和对正例的覆盖能力之间的权衡。对于随机分类器而言，其p固定的等于样本中正例的比例，不随r的变化而变化。
多分类中每个类别可以绘制出一条pr曲线,曲线中为依次改变置信度为10％-100％得到一组由p和r组成的坐标,连接这些值就是pr曲线。
第二平均精度，它被定义为查准率和召回率的调和平均数。
一般在查全率和查准率上有偏向的时候,会第二平均精度,例如推荐系统中，如果需要更精准的了解客户需求，避免推送用户不感兴趣的内容，p就更加重要；在疾病检测的时候，我们不需要查漏任何一项疾病，这时r(tpr)就更重要。当两者都需要考虑时，第二平均精度就是一种参考指标。
如表1所示，为增加图像分割网路和未增加图像分割网络的第一平均金谷地和第二平均精度的对比。
表1
如图7所示，为添加图像分割网络前后的特征图对比图。
可见，将图像分割网络和落石检测网络结合后，第一平均精度和第二平均精度均提高。
在本申请的另一个实施例中，如图8所示，提供了一种落石检测方法，具体包括以下步骤：
步骤401：将目标图像输入至图像分割网络。
步骤402：图像分割网络检测目标图像中的区域标识。
其中，区域标识指示目标图像的高速道路区域和非高速道路区域。
步骤403：将携带有区域标识输入的目标图像输入至落石检测网络中。
步骤404：落石检测网络对目标图像以及目标图像对应的标识信息进行检测。
步骤405：落石检测网络输出高速道路区域的落石信息和非高速道路的落石信息。
将目标图像通过图像分割网络和落石检测网络，将图像先输入至图像分割网络中，确定目标图像中高速道路区域和非高速道路区域，将标注好区域的目标图像，即携带有区域标识输入的目标图像作为落石检测网络的输入，得到高速道路区域和非高速道路区域的落石信息。在对高速道路落石提前检测的条件下，可提前发出预警，提醒驾驶者注意道路安全或在道路堵塞的情况下选择其他道路。
如图9所示，基于训练后的模型的检测效果图。
采用训练好的模型对目标图像中的落石信息进行检测，把注意力放在高速道路上，从而减少高速道路外的误判情况。
请参照图10，本申请实施例还提供了一种应用于图1所述电子设备100的落石检测模型训练装置110，所述落石检测模型训练装置110包括：
第一计算模块111，用于计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息，其中，所述目标图像的区域标识由图像分割网络输出，所述标准图像的区域标识由人工标注，所述区域标识指示所述目标图像或所述标准图像中的高速道路区域和非高速道路区域，所述第一损失惩罚信息为目标图像中的区域标识与标准图像中的区域标识的差异。
在本实施例中，所述第一计算模块111可用于执行图2所示的步骤201，关于所述第一计算模块的具体描述可参照对所述步骤201的描述。
第二计算模块112，用于计算所述目标图像中目标落石信息与所述标准图像中标准落石信息的第二损失惩罚项信息，其中，所述第二损失惩罚项信息为目标图像中的落石信息与所述标准图像的标准落石信息的差异。
在本实施例中，所述第二计算模块112可用于执行图2所示的步骤202，关于所述第二计算模块的具体描述可参照对所述步骤202的描述。
输入模块113，用于将所述第一损失惩罚项信息和所述损失第二惩罚项信息分别输入至图像分割网络和落石检测网络中，得到更新后的目标图像和更新后的所述目标图像的目标落石信息。
在本实施例中，所述输入模块113可用于执行图2所示的步骤203，关于所述输入模块的具体描述可参照对所述步骤203的描述。
执行模块114，用于返回执行所述计算目标图像和标准图像的第一损失惩罚项信息的步骤，直至最新的第一惩罚项信息和最新的第二惩罚项信息满足落石检测条件。
在本实施例中，所述执行模块114可用于执行图2所示的步骤204，关于所述执行模块的具体描述可参照对所述步骤204的描述。
综上所述，本申请通过计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息，计算目标图像中目标落石信息与标准图像中标准落石信息的第二损失惩罚项信息；将第一损失惩罚项信息和损失第二惩罚项信息分别输入至图像分割网络和落石检测网络中，得到更新后的目标图像的区域标识和更新后的目标图像的目标落石信息；返回执行计算目标图像的区域标识和标准图像的区域标识的第一损失惩罚项信息的步骤，直至最新的第一惩罚项信息和最新的第二惩罚项信息满足落石检测条件。本申请通过图像分割网络，将落石的注意力放在高速道路上，从而减少高速道路外的误判情况。高速道路的落石检测对于提高山区驾驶安全有着重要作用。
请参照图11，本申请实施例还提供了一种落石检测设备，所述设备包括：
第一输入模块501，用于将目标图像输入至所述的图像分割网络；
第一检测模块502，用于所述图像分割网络检测所述目标图像中的区域标识，其中，所述区域标识指示所述目标图像的高速道路区域和非高速道路区域；
第二输入模块503，用于将携带有区域标识输入的所述目标图像输入所述的落石检测网络中；
第二检测模块504，用于所述落石检测网络对所述目标图像以及所述目标图像对应的标识信息进行检测；
输出模块505，用于所述落石检测网络输出高速道路区域的落石信息和非高速道路的落石信息。
本申请通过将目标图像通过图像分割网络和落石检测网络，将图像先输入至图像分割网络中，确定目标图像中高速道路区域和非高速道路区域，将标注好区域的目标图像，即携带有区域标识输入的目标图像作为落石检测网络的输入，得到高速道路区域和非高速道路区域的落石信息。在对高速道路落石提前检测的条件下，可提前发出预警，提醒驾驶者注意道路安全或在道路堵塞的情况下选择其他道路。
本申请还提供一种电子设备100，电子设备100包括处理器130以及存储器120。存储器120存储有计算机可执行指令，计算机可执行指令被处理器130执行时，实现该落石检测模型训练方法。
本申请实施例还提供一种计算机可读存储介质，存储介质存储有计算机程序，计算机程序被处理器130执行时，实现该落石检测模型训练方法。
在本申请所提供的实施例中，应该理解到，所揭露的装置和方法，也可以通过其它的方式实现。以上所描述的装置实施例仅仅是示意性的，例如，附图中的流程图和框图显示了根据本申请的多个实施例的装置、方法和计算机程序产品的可能实现的体系架构、功能和操作。在这点上，流程图或框图中的每个方框可以代表一个模块、程序段或代码的一部分，所述模块、程序段或代码的一部分包含一个或多个用于实现规定的逻辑功能的可执行指令。也应当注意，在有些作为替换的实现方式中，方框中所标注的功能也可以以不同于附图中所标注的顺序发生。例如，两个连续的方框实际上可以基本并行地执行，它们有时也可以按相反的顺序执行，这依所涉及的功能而定。也要注意的是，框图和/或流程图中的每个方框、以及框图和/或流程图中的方框的组合，可以用执行规定的功能或动作的专用的基于硬件的系统来实现，或者可以用专用硬件与计算机指令的组合来实现。
另外，在本申请各个实施例中的各功能模块可以集成在一起形成一个独立的部分，也可以是各个模块单独存在，也可以两个或两个以上模块集成形成一个独立的部分。所述功能如果以软件功能模块的形式实现并作为独立的产品销售或使用时，可以存储在一个计算机可读取存储介质中。基于这样的理解，本申请的技术方案本质上或者说对现有技术做出贡献的部分或者该技术方案的部分可以以软件产品的形式体现出来，该计算机软件产品存储在一个存储介质中，包括若干指令用以使得一台计算机设备(可以是个人计算机，服务器，或者网络设备等)执行本申请各个实施例所述方法的全部或部分步骤。而前述的存储介质包括：u盘、移动硬盘、只读存储器(rom，read-onlymemory)、随机存取存储器(ram，randomaccessmemory)、磁碟或者光盘等各种可以存储程序代码的介质。
需要说明的是，在本文中，诸如第一和第二等之类的关系术语仅仅用来将一个实体或者操作与另一个实体或操作区分开来，而不一定要求或者暗示这些实体或操作之间存在任何这种实际的关系或者顺序。而且，术语“包括”、“包含”或者其任何其他变体意在涵盖非排他性的包含，从而使得包括一系列要素的过程、方法、物品或者设备不仅包括那些要素，而且还包括没有明确列出的其他要素，或者是还包括为这种过程、方法、物品或者设备所固有的要素。在没有更多限制的情况下，由语句“包括一个……”限定的要素，并不排除在包括所述要素的过程、方法、物品或者设备中还存在另外的相同要素。
以上所述，仅为本申请的各种实施方式，但本申请的保护范围并不局限于此，任何熟悉本技术领域的技术人员在本申请揭露的技术范围内，可轻易想到变化或替换，都应涵盖在本申请的保护范围之内。因此，本申请的保护范围应所述以权利要求的保护范围为准。
