公开(公告)号：CN116258937A

公开(公告)日：2023-06-13

申请号：CN202211569888.1

申请日：2022-12-08

Applicant: 鹏城实验室

Inventor： 陈杰 ,  田永鸿 ,  高文 ,  王中岐 ,  任前

IPC: G06V10/82 ,  G06V10/774 ,  G06V10/764 ,  G06V10/10 ,  G06V10/26 ,  G06N3/045 ,  G06N3/09

Abstract: 本发明公开了基于注意力机制的小样本分割方法、装置、终端及介质，包括：输入支持图片、支持图片对应的掩码图以及待预测查询图片；通过权重共享的编码器分别提取待预测查询图片的特征和支持图片的特征；将支持图片的特征和支持图片对应的掩码图输入原型生成算法，得到一对互补的原型；将一对互补的原型扩展至待预测查询图片的尺寸，并与待预测查询图片进行拼接，通过FPN结构的解码器预测拼接后的前后背景，得到分割结果；根据双线性插值算法将分割结果恢复至原始图像尺寸大小，得到小样本分割结果。本发明以一种权重互补的方式构建一对原型，从而在原型生成过程中尽量地保留的有效信息，减少原型偏置造成的分类错误，提高判断精度。