CN114819100B 目标检测模型的神经网络搜索方法、装置和电子设备 （北京百度网讯科技有限公司）

本公开提供了一种目标检测模型的神经网2获取搜索空间，所述搜索空间包括多个网络模块，所述多个网络对所述搜索空间当前可搜索的网络模块中进行搜索，获取用将所述可搜索的网络模块中剩余的第二网络模块通过设定基于所述第一网络模块和所述第二网络模块对应的所述设定卷积层针对任一候选检测模型，基于第一训练样本图像对所述候选检测选取符合设定条件的训练后候选检测模型作为目标检从多个所述训练后候选检测模型中，按照进化算法选取待验证的训练后候选检测模基于验证样本图像对所述待验证的训练后候选检测模型进行精度验选取模型性能参数和所述检测模型精度满足条件的所述待验证的训练后候选检测模基于第二训练样本图像对所述目标检测模型进行重新训练，以获取针对第i次所述神经网络结构搜索，确定前面i_1次在所述基于所述已搜索过的网络模块，确定所述搜索空间中剩余的基于所述第二训练样本图像对删除所述设定卷积层的所述目标检测模型进行重新训在所述神经网络结构搜索过程中为所述第一网络3搜索模块，用于对所述搜索空间中的所述网络模块进行多次神经网络训练模块，用于针对任一候选检测模型，基于第一训对所述搜索空间当前可搜索的网络模块中进行搜索，获取用于构成将所述可搜索的网络模块中剩余的第二网络模块通过设定基于所述第一网络模块和所述第二网络模块对应的所述设定卷积层从多个所述训练后候选检测模型中，按照进化算法选取待验证的训练后候选检测模基于验证样本图像对所述待验证的训练后候选检测模型进行精度验选取模型性能参数和所述检测模型精度满足条件的所述待验证的训练后候选检测模基于第二训练样本图像对所述目标检测模型进行重新训练，以获取针对第i次所述神经网络结构搜索，确定前面i_1次在所述基于所述已搜索过的网络模块，确定所述搜索空间中剩余的基于所述第二训练样本图像对删除所述设定卷积层的所述目标检测模型进行重新训在所述神经网络结构搜索过程中为所述第一网络4所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被理器执行，以使所述至少一个处理器能够执行权利要求1_7中任一项所述的目标检测模型使所述计算机执行根据权利要求1_7中任一项所述的目标检测模型的据权利要求1_7中任一项所述的目标检测模型5[0003]然而在计算资源的限制下，人为设计检测模型只能依据[0007]对搜索空间中的网络模块进行多次神经网络结构搜索，6使至少一个处理器能够执行上述一方面实施例的目标检测模型的序被处理器执行时实现上述一方面实施例的目标检测模[0021]应当理解，本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特[0023]图1为本公开实施例提供的一种目标检测模型的神经网络搜索方法的流程示意[0024]图2为本公开实施例提供的一种目标检测模型的神经网络搜索方法的搜索空间的[0025]图3为本公开实施例提供的另一种目标检测模型的神经网络搜索方法的流程示意[0026]图4为本公开实施例提供的另一种目标检测模型的神经网络搜索方法的流程示意[0027]图5为本公开实施例提供的另一种目标检测模型的神经网络搜索方法的流程示意[0028]图6为本公开实施例提供的另一种目标检测模型的神经网络搜索方法的流程示意[0029]图7为本公开实施例提供的一种目标检测模型的神经网络搜索方法的采样样本的[0030]图8为本公开实施例提供的一种目标检测模型的神经网络搜索装置的结构示意[0031]图9为根据本公开实施例的目标检测模型的神经网络搜索方法的电子设备的框7[0039]图1为本公开实施例提供的一种目标检测模型的神经网络搜索方法的流程示意[0041]搜索空间包括多层神经网络，各层网络存在对应的搜索神经网络的网络模块集实施例提出了一种搜索空间，这种搜索空间针对检测模型的block和channel进行不同组度和宽度的检测模型。本公开方案中提出的搜索空间可以实现自动搜索来设计检测模型，8三部分所占比重应该根据具体情况来适当修改才能使检测网络实检测颈部网络和检测头部网络进行分别解耦，这种搜索出的检索模型具有更高的一致性，术在搜空间中构建多个候选检测模型，需要说明的是，每个候选检测模型的block数和9[0066]S303，选取模型性能参数和模型精度满足条件的待验证[0071]图4为本公开实施例提供的另一种目标检测模型的神经网络搜索方法的流程示意得到的候选检测模型包括block1和block2，第二次搜索需要在block3……blocki……blockN等N_2个网络模块中确定下一个候索空间的采样采用不放回采样的方式即可保证搜索[0095]在本公开实施例中，还需要基于第二训练样本图像对目标检测模型进行重新训检测模型的神经网络搜索装置与上述几种实施例提供的目标检测模型的神经网络搜索方[0102]图8为本公开实施例提供的一种目标检测模型的神经网络搜索装置的结构示意的计算机程序或者从存储单元906载到随机访问存储器(RAM)903中的计算机程序，来执行特网的计算机网络和/或各种电信网络与其他设机程序可在包括至少一个可编程处理器的可编程系统上执行和/或解释，该可编程处理器[0123]用于实施本公开的方法的程序代码可以采用一个或多个编程语言的任何组合来理器或控制器，使得程序代码当由处理器或控制器执行时使流程图和/或框图中所规定的包部分地在机器上执行且部分地在远程机器上执行或完全在远程机器或服务器面或者该网络浏览器来与此处描述的系统和技术的实施方式交互)、或者包括这种后台部字数据通信(例如，