神经网络训练与构建方法和装置以及目标检测方法和装置

摘要

本发明的实施例提供了用于目标检测的神经网络训练与构建方法和装置以及基于神经网络的目标检测方法和装置。所述用于目标检测的神经网络训练方法包括：将包含有训练目标的训练图像输入至神经网络，以得到所述训练目标的预测边界框；根据所述预测边界框和真实边界框的交集面积与并集面积之比，获得第一损失函数，所述真实边界框为在所述训练图像中预先标注的该训练目标的边界框；以及至少利用所述第一损失函数调整所述神经网络的参数，以对所述神经网络进行训练。本发明实施例的神经网络训练与构建方法和装置以及目标检测方法和装置利用第一损失函数将目标边界框回归为一个整体单元，显著增加了神经网络的目标检测精度，并且能够通过神经网络的两个分支结构有效提高神经网络的训练和检测效率。

主权项

一种用于目标检测的神经网络的训练方法，包括：将包含有训练目标的训练图像输入至神经网络，以得到所述训练目标的预测边界框；根据所述预测边界框和真实边界框的交集面积与并集面积之比，获得第一损失函数，所述真实边界框为在所述训练图像中预先标注的该训练目标的边界框；以及至少利用所述第一损失函数调整所述神经网络的参数，以对所述神经网络进行训练。