[发明专利]用于设备缺陷的多阶段识别方法及系统

权利要求书

1.一种用于设备缺陷的多阶段识别方法，其特征在于，所述多阶段识别方法包括：

获取待识别的设备图像；

将所述设备图像输入卷积单元中，以得到第一特征图；

将所述第一特征图输入区域生成网络，以得到多个候选区域；

将所述第一特征图和多个所述候选区域输入感兴趣池化单元中，以得到候选特征图；

将所述候选特征图输入全连接层，以得到所述候选区域的类别；

将所述候选区域以及对应的类别输入边界框回归单元，以进行边界修正。

2.根据权利要求1所述的多阶段识别方法，其特征在于，获取待识别的设备图像包括：

将所述设备图像进行预处理，并缩放至预设的尺寸。

3.根据权利要求1所述的多阶段识别方法，其特征在于，将所述设备图像输入卷积单元中，以得到第一特征图包括：

对所述第一特征图依次执行四次卷积操作、激活操作、卷积操作、激活操作、池化操作，以得到第二特征图；

对所述第二特征图依次执行卷积操作、激活操作、卷积操作、激活操作以及卷积操作，以得到所述第一特征图。

4.根据权利要求1所述的多阶段识别方法，其特征在于，所述卷积操作的参数为kernel_size＝3，pad＝1，stride＝1。

5.根据权利要求1所述的多阶段识别方法，其特征在于，所述池化操作的参数为kernel_size＝2，pad＝0，stride＝2。

6.根据权利要求1所述的多阶段识别方法，其特征在于，将所述第一特征图输入区域生成网络，以得到多个候选区域包括：

对于所述第一特征图依次执行卷积操作和激活操作，以得到第三特征图；

对所述第三特征图执行维度调整操作；

将维度调整操作后的所述第三特征图输入softmax层，以得到多个候选框；

将所述候选框和所述第三特征图进行叠加，以得到多个所述候选区域。

7.根据权利要求1所述的多阶段识别方法，其特征在于，将所述候选特征图输入全连接层，以得到所述候选区域的类别包括：

将所述候选区域和所述第一特征图输入全连接层中，以得到每个候选区域的类别以及对应的概率；

对包括候选区域的特征图执行依次全连接操作、激活操作、全连接操作以及激活操作，以得到第四特征图；

采样softmax函数根据所述第四特征图进行分类，以得到每个所述候选区域的类别以及对应的概率。

8.一种用于设备缺陷的多阶段识别系统，其特征在于，所述识别系统包括处理器，所述处理器用于执行如权利要求1至7任一所述的多阶段识别方法。

9.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质存储有指令，所述指令用于被机器读取以使得所述机器执行如权利要求1至7任一所述的多阶段识别方法。