[发明专利]基于NAS网络的目标检测方法及系统

权利要求书

1.一种基于NAS网络的目标检测方法，其特征在于，包括：

通过FPN网络对图像进行处理，生成所述图像的第一特征图；

通过FPN网络对所述第一特征图自上而下进行特征融合，得到第二特征图，直至所述第二特征图形成第一特征金字塔；其中，所述第二特征图包括激活层、卷积层和归一化层；

将所述第一特征金字塔输入到NAS网络中，所述NAS网络包括至少两个第一连接模式，根据每个第一连接模式对所述第一特征金字塔中的任意两个特征进行融合得到至少两个第二特征金字塔；

判断所述第二特征金字塔的性能是否达到要求；

若所述第二特征金字塔的性能达到要求，则选择所述第二特征金字塔对应的第一连接模式；

根据所述NAS网络对选择的所述第一连接模式进行参数调节，得到第二连接模式；

根据所述第二连接模式对所述第二特征金字塔中的任意两个特征进行特征融合得到第三特征金字塔；

将所述第三特征金字塔投入到分类器进行图像分类识别，得到第三特征图，对所述第三特征进行边界框回归处理，完成目标检测。

2.如权利要求1所述的基于NAS网络的目标检测方法，其特征在于，所述判断所述第二特征金字塔的性能是否达到要求，包括：

判断所述第二特征金字塔的性能是否优于所述第一特征金字塔的性能，若所述第二特征金字塔的性能优于所述第一特征金字塔的性能，则所述第二特征金字塔的性能达到要求；

或，判断所述第二特征金字塔的性能是否超过第一阈值，若第二特征金字塔的性能超过所述第一阈值，则所述第二特征金字塔的性能达到要求。

3.如权利要求1或2所述的基于NAS网络的目标检测方法，其特征在于，根据所述第二连接模式对所述第二特征金字塔中的任意两个特征进行特征融合得到第三特征金字塔后，还包括：判断所述第三特征金字塔的性能是否达到要求，若达到要求则选择所述第三特征金字塔，并将所述第三特征金字塔投入到分类器进行图像分类识别；

其中，所述判断所述第三特征金字塔的性能是否达到要求包括：

判断所述第三特征金字塔的性能是否优于所述第二特征金字塔的性能，若所述第三特征金字塔的性能优于所述第二特征金字塔的性能，则所述第三特征金字塔的性能达到要求；

或，判断所述第三特征金字塔的性能是否超过第二阈值，若第三特征金字塔的性能超过所述第二阈值，则所述第三特征金字塔的性能达到要求。

4.一种基于NAS网络的目标检测系统，其特征在于，包括FPN网络、NAS网络和分类器，所述FPN网络包括：

特征生成单元，对图像进行处理，生成所述图像的第一特征图；

特征融合单元，对所述第一特征图自上而下进行特征融合，得到第二特征图，直至所述第二特征图形成第一特征金字塔；其中，所述第二特征图包括激活层、卷积层和归一化层；

所述NAS网络包括：

第一特征融合单元，包括至少两个第一连接模式，根据每个第一连接模式对所述第一特征金字塔中的任意两个特征进行融合得到至少两个第二特征金字塔；

第一判断单元，判断所述第二特征金字塔的性能是否达到要求；

第一选择单元，若所述第二特征金字塔的性能达到要求，则选择所述第二特征金字塔对应的第一连接模式；

参数调节单元，根据所述NAS网络对选择的所述第一连接模式进行参数调节，得到第二连接模式；

第二特征融合单元，根据所述第二连接模式对所述第二特征金字塔中的任意两个特征进行特征融合得到第三特征金字塔；

所述分类器对所述第三特征金字塔进行图像分类识别，得到第三特征图，对所述第三特征进行边界框回归处理，完成目标检测。

5.如权利要求4所述的基于NAS网络的目标检测系统，其特征在于，所述第一判断单元还用于：

判断所述第二特征金字塔的性能是否优于所述第一特征金字塔的性能，若所述第二特征金字塔的性能优于所述第一特征金字塔的性能，则所述第二特征金字塔的性能达到要求；

或，判断所述第二特征金字塔的性能是否超过第一阈值，若第二特征金字塔的性能超过所述第一阈值，则所述第二特征金字塔的性能达到要求。

6.如权利要求4或5所述的基于NAS网络的目标检测系统，其特征在于，该系统还包括第二判断单元和第二选择单元，所述第二判断单元用于判断所述第三特征金字塔的性能是否达到要求，若达到要求则所述第二选择单元选择所述第三特征金字塔，并将所述第三特征金字塔投入到分类器进行图像分类识别；

所述第二判断单元还用于：

判断所述第三特征金字塔的性能是否优于所述第二特征金字塔的性能，若所述第三特征金字塔的性能优于所述第二特征金字塔的性能，则所述第三特征金字塔的性能达到要求；

或，判断所述第三特征金字塔的性能是否超过第二阈值，若第三特征金字塔的性能超过所述第二阈值，则所述第三特征金字塔的性能达到要求。