[发明专利]目标检测方法及装置、电子设备和存储介质

说明书

本公开涉及一种目标检测方法及装置、电子设备和存储介质，其中，所述方法包括：获取目标图像；利用目标检测网络对所述目标图像进行检测，得到所述目标图像中目标对象的第一检测结果；其中，所述目标检测网络是基于样本图像以及样本图像的监督信息训练得到的，以及所述样本图像的监督信息包括掩膜图。本公开实施例可以应用于医疗场景中，可对医学影像进行目标检测，提高目标检测网络的检测精度。

技术领域

本公开涉及计算机视觉技术领域，尤其涉及一种目标检测方法及装置、电子设备和存储介质。

背景技术

计算机视觉技术是利用计算机以及相关设备对生物视觉进行模拟的一种技术，通过对采集的图像或视频进行处理可以获得相应场景的信息，从而模拟生物的观测。目标检测是计算视觉技术的一个重要任务。

目标检测可以应用在多种场景，例如，在医学场景中，可以利用目标检测对医学影像的病灶区域进行识别，对医学影像辅助分析。但是目前目标检测任务中，检测结果的精度较低。

发明内容

本公开提出了一种目标检测技术方案。

根据本公开的一方面，提供了一种目标检测方法，包括：

获取目标图像；利用目标检测网络对所述目标图像进行检测，得到所述目标图像中目标对象的第一检测结果；其中，所述目标检测网络是基于样本图像以及样本图像的监督信息训练得到的，以及所述样本图像的监督信息包括掩膜图。

在本公开实施例中，可以获取待检测的目标图像，利用目标检测网络对目标图像进行检测，得到目标图像中目标对象的第一检测结果。这里，目标检测网络是基于样本图像以及样本图像的监督信息得到的，样本图像的监督信息包括掩膜图，掩膜图可以对样本图像对应的空间信息进行监督，从而由样本图像以及样本图像的监督信息得到的目标检测网络可以保留目标图像的空间信息，提高目标检测网络的检测精度。

在一个或多个可能的实现方式中，所述利用目标检测网络对所述目标图像进行检测，得到所述目标图像中目标对象的第一检测结果，包括：基于所述目标检测网络，确定至少一个第一候选区域；基于所述至少一个第一候选区域的图像特征，得到包括所述目标对象的类别信息和位置信息的第一检测结果。

通过目标检测网络可以确定目标对象可能位于的至少一个第一候选区域，可以针对第一候选区域中可能存在的目标对象进行检测，从而使检测到的目标对象更加全面。

在一个或多个可能的实现方式中，所述方法还包括：通过一神经网络对所述样本图像进行处理，得到所述样本图像中参照对象的第二检测结果和热力图；基于所述第二检测结果、所述热力图以及所述样本图像的监督信息，确定所述神经网络的网络损失，其中，所述热力图用于指示所述参照对象在所述样本图像中的预测位置；基于所述网络损失对所述神经网络的网络参数进行调整，得到训练好的神经网络，其中，所述训练好的神经网络包括所述目标检测网络。

这里，由于神经网络在训练过程中对提供空间信息的热力图进行监督，从而由训练好的神经网络得到的目标检测网络可具有较高的检测精度。

在一个或多个可能的实现方式中，所述神经网络包括检测分支和分割分支；所述通过一神经网络对所述样本图像进行处理，得到所述样本图像中所述参照对象的第二检测结果和热力图，包括：利用所述神经网络对所述样本图像进行特征提取，得到至少一个第二候选区域；将所述至少一个第二候选区域输入所述检测分支，得到所述样本图像中参照对象的第二检测结果；将所述至少一个第二候选区域输入所述分割分支，得到所述参照对象的热力图。

这里，由于神经网络的检测分支和分割分支共享样本图像的图像特征，分割分支可以保留特征空间的结构信息，从而可以提高检测分支的检测精度。分割分支采用放射状的掩膜图进行监督，相比于精细标注对象轮廓的完整掩膜图而言，可以大大减少人工标注成本。