【技术实现步骤摘要】
目标检测模型的神经网络搜索方法、装置和电子设备
[0001]本公开涉及人工智能
，具体涉及智能搜索、深度学习、图像处理和计算机视觉
，尤其涉及一种目标检测模型的神经网络搜索方法、装置和电子设备及存储介质。
技术介绍
[0002]深度学习是机器学习的一个分支，目的在于建立一个模拟人脑进行分析学习的神经网络，通过模仿人脑的工作机制来进行数据处理，例如图像通常应用在视频识别、图像识别或者声音识别领域。
[0003]然而在计算资源的限制下，人为设计检测模型只能依据以往不断的实验与经验，无法对无尽的检测模型进行实验。目标检测模型依赖于骨干检测模型来学习图像特征，因此骨干检测模型的效果对目标检测模型的影响很大。然而由于学习任务不同，很难说最优的骨干检测模型在检测任务里也能带来最优的效果。
技术实现思路
[0004]本公开提供了一种用于目标检测模型的神经网络搜索方法、装置、电子设备及存储介质。
[0005]根据本公开的一方面，提供了一种目标检测模型的神经网络搜索方法，包括：
[0006]...
【技术保护点】
【技术特征摘要】
1.一种目标检测模型的神经网络搜索方法，包括：获取搜索空间，所述搜索空间包括多个网络模块，所述多个网络模块用于组合成骨干网络、检测颈部网络和检测头部网络；对所述搜索空间中的所述网络模块进行多次神经网络结构搜索，得到多个候选检测模型，其中，所述候选检测模型包括候选骨干网络、候选检测颈部网络和候选检测头部网络；针对任一候选检测模型，基于第一训练样本图像对所述候选检测模型进行训练，以对所述候选检测模型进行模型调整，得到训练后候选检测模型；选取符合设定条件的训练后候选检测模型作为目标检测模型。2.根据权利要求1所述的方法，其中，所述选取符合设定条件的训练后候选检测模型作为目标检测模型，包括：从多个所述训练后候选检测模型中，按照进化算法选取待验证的训练后候选检测模型；基于验证样本图像对所述待验证的训练后候选检测模型进行精度验证，获取所述待验证的训练后候选检测模型的模型精度；选取模型性能参数和所述检测模型精度满足条件的所述待验证的训练后候选检测模型，作为所述目标检测模型。3.根据权利要求1所述的方法，其中，所述目标检测模型确定之后，还包括：基于第二训练样本图像对所述目标检测模型进行重新训练，以获取最终的目标检测模型。4.根据权利要求1所述的方法，其中，所述方法还包括：针对第i次所述神经网络结构搜索，确定前面i
‑
1次在所述搜索空间内已搜索过的网络模块，其中，所述i为大于1的正整数，且不大于设定的总搜索次数；基于所述已搜索过的网络模块，确定所述搜索空间中剩余的所述网络模块，并在所述剩余的网络模块中进行搜索。5.根据权利要求1
‑
4中任一项所述的方法，其中，在所述搜索空间中进行所述神经网络结构搜索，得到所述候选检测模型的过程，包括：对所述搜索空间当前可搜索的网络模块中进行搜索，获取用于构成所述候选检测模型的第一网络模块；将所述可搜索的网络模块中剩余的第二网络模块通过设定卷积层进行替代；基于所述第一网络模块和所述第二网络模块对应的所述设定卷积层，生成所述候选检测模型。6.根据权利要求5所述的方法，其中，所述基于第二训练样本图像对所述目标检测模型进行重新训练，包括：删除所述目标检测模型中所述设定卷积层；基于所述第二训练样本图像对删除所述设定卷积层的所述目标检测模型进行重新训练。7.根据权利要求5所述的方法，其中，所述网络模块包括多个通道，所述多个通道共享最大通道数的权重，所述方法还包括：在所述神经网络结构搜索过程中为所述第一网络模块随机选取通道；
在训练时所述选取通道共享所述最大通道数的权重进行训练。8.根据权利要求7所述的方法，其中，对所述候选模型进行模型调整，包括：响应于选取的通道数小于所述最大通道数，从所述最大通道数的权重中切分出所述随机选取通道对应的权重；对所述随机选取通道对应的权重进行梯度更新。9.一种目标检测模型的神经网络搜索装置，包括：获取模块，用于获取搜索空间，所述搜索空间包括多个网络模块，所述多个网络模块用于组合成骨干网络、检测颈部网络和检测头部网络；搜索模块，用于对所述搜索空间中的所述网络模块进行多次神经网络结构搜索，得到多个候选检测模型，其...
【专利技术属性】
技术研发人员：徐畅，于广华，王豪爽，党青青，李明昊，刘其文，
申请(专利权)人：北京百度网讯科技有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。