[发明专利]基于目标检测模型的腹腔镜手术组织器官识别方法及装置

权利要求书

1.一种基于目标检测模型的腹腔镜手术组织器官识别方法，包括：

提取输入图像的第一特征图；

对所述第一特征图进行局部信息静态建模和上下文信息动态建模，获取局部上下文信息增强的增强特征图；

基于所述增强特征图进行目标检测，得到所述输入图像中包含的器官信息及相应的标识；

对所述第一特征图进行局部信息静态建模，包括：

采用特定的第一卷积核对所述第一特征图进行卷积，以对所述第一特征图进行局部信息静态建模，得到具备局部上下文信息表示的静态注意力键值；对所述第一特征图进行上下文信息动态建模，包括：

采用特定的第二卷积核对所述第一特征图进行卷积，得到所述第一特征图的特征值，以便对所述第一特征图进行上下文信息动态建模；

基于预先构建的记忆力单元对所述第一特征图进行点乘，得到注意力的查询信息，其中，所述记忆力单元的尺寸与所述第一特征图的尺寸相同；

将所述查询信息与所述静态注意力键值进行计算，得到自注意力增强特征图；

基于所述特征值和所述自注意力增强特征图进行点乘计算，得到上下文信息动态建模的特征图。

2.根据权利要求1所述的基于目标检测模型的腹腔镜手术组织器官识别方法，将所述查询信息与所述静态注意力键值进行计算，得到自注意力增强特征图，包括：

将所述查询信息与所述静态注意力键值进行拼接，得到拼接后的第二特征图；

基于多层感知机对所述拼接后的第二特征图进行计算，得到自注意力增强特征图。

3.根据权利要求1或2所述的基于目标检测模型的腹腔镜手术组织器官识别方法，对所述第一特征图进行局部信息静态建模和上下文信息动态建模，获取局部上下文信息增强的增强特征图，包括：

将所述静态注意力键值和所述上下文信息动态建模的特征图相加，得到局部上下文信息增强的增强特征图。

4.根据权利要求1所述的基于目标检测模型的腹腔镜手术组织器官识别方法，基于所述增强特征图进行目标检测，得到所述输入图像中包含的器官信息及相应的标识，包括：

基于特征金字塔网络对所述增强特征图进行特征融合，得到至少一个待识别特征图；

对所述待识别特征图进行目标检测，确定所述待识别特征图中包含的目标；

识别所述目标对应的器官信息。

5.根据权利要求4所述的基于目标检测模型的腹腔镜手术组织器官识别方法，所述器官信息中包括器官名称和识别概率，识别所述目标对应的器官信息之后，所述方法还包括：

在所述输入图像中输出用于标识目标所在位置的包围框；

在所述输入图像中的包围框内输出与所述包围框所标识的目标对应的器官名称和识别概率。

6.一种存储有程序的存储介质，其中，所述存储介质存储有计算机程序，所述计算机程序被处理器执行时实现如权利要求1~5中的任一项所述的方法。

7.一种计算设备，包括如权利要求6所述的存储介质。