[发明专利]一种处理装置、方法及存储介质

权利要求书

1.一种处理装置，其特征在于，所述处理装置包括采集模块以及训练模块，所述训练模块包括主干网络以及候选区域生成网络RPN层，所述主干网络与所述RPN层连接，所述RPN层包括类激活热力图CAM单元，

所述采集模块，用于获取图像，所述图像包括实例级标注的图像以及图像级标注的图像，所述实例级标注的图像包括类别标签和物体边框，所述图像级标注的图像包括类别标签；

所述主干网络，用于根据所述采集模块获取的所述图像，输出所述图像的特征图；

所述RPN层，用于根据所述特征图确定所述图像的候选区域；

所述RPN层，还用于通过所述CAM单元确定所述候选区域对应的CAM，并根据所述CAM确定所述候选区域属于前景的概率。

2.根据权利要求1所述的处理装置，其特征在于，所述采集模块，具体用于：

接收实例级标注的图像；

根据所述实例级标注的图像包括的所述类标签获取所述图像级标注的图像。

3.根据权利要求1或2所述的处理装置，其特征在于，所述CAM指示所述候选区域的CAM响应强度，所述RPN层，具体用于：

根据所述候选区域的所述CAM响应强度确定所述候选区域属于前景的概率，所述候选区域的所述CAM响应强度和所述候选区域属于前景的概率成正比。

4.根据权利要求1或2所述的处理装置，其特征在于，所述CAM指示所述候选区域的CAM响应强度，所述RPN层，具体用于：

根据所述候选区域的置信度和所述CAM响应强度的加权和确定所述候选区域作为前景的概率。

5.根据权利要求1至4任一项所述的处理装置，其特征在于，所述处理装置还包括N个头部head，所述N为正整数，所述head与所述RPN层连接，所述N个head中的任意一个用于：

根据所述候选区域的特征确定所述实例级标注的图像以及所述图像级标注的图像的多示例检测MID损失，所述候选区域的特征为所述RPN层根据所述前景的概率的大小确定的候选区域对应的特征，所述MID损失用于确定所述训练模块的参数。

6.根据权利要求5所述的处理装置，其特征在于，

所述N个head串联。

7.根据权利要求5或6所述的处理装置，其特征在于，所述RPN层，还用于：

计算所述实例级标注图像与所述图像级标注图像的CAM损失，所述CAM损失用于确定所述训练模块的参数。

8.根据权利要求7所述的处理装置，其特征在于，

所述RPN层，还用于：

计算所述实例级标注图像的RPN损失；

所述head，还用于：

计算所述实例级标注图像的候选损失、分类损失以及回归损失，所述RPN损失、所述候选损失、所述分类损失以及所述回归损失用于确定所述训练模块的参数。

9.根据权利要求1至7任一项所述的处理装置，其特征在于，所述处理装置为云服务器。

10.一种物体检测装置，其特征在于，所述物体检测装置包括主干网络、候选区域生成网络RPN层，所述主干网络与所述RPN层连接，所述RPN层包括类激活热力图CAM单元，

所述主干网络，用于接收待检测图像，输出所述待检测图像的特征图；

所述RPN层，用于根据所述特征图确定所述待检测图像的候选区域，并通过所述CAM单元确定所述候选区域对应的CAM；

所述RPN层，还用于根据所述CAM确定所述候选区域属于前景的概率。

11.根据权利要求10所述的物体检测装置，其特征在于，所述CAM指示所述候选区域的CAM响应强度，所述RPN层，具体用于：

根据所述候选区域的所述CAM响应强度确定所述候选区域属于前景的概率，所述候选区域的所述CAM响应强度和所述候选区域属于前景的概率成正比。