一种基于掩膜编码的单阶段实例分割方法

摘要

本发明涉及图像处理领域，特别涉及一种基于掩膜编码的单阶段实例分割方法，包括训练阶段模型和预测阶段模型，其特征在于，按照以下步骤进行：步骤A，双线性插值：将不同尺寸给定的原始掩膜标签去掉类别信息，得到尺寸相同且类别无关的二维掩膜；步骤B，主成分分析编码：根据给定信息冗余的二维掩码，通过正交变换将二维掩膜M编码压缩为向量uN；步骤C，单阶段实例分割器构建：单阶段实例分割器基于单阶段目标检测器FCOS，并进行相应修改；步骤D，非极大值抑制后处理；步骤E，掩膜编码解码。本发明将高度冗余的二维掩膜压缩至特征紧凑的掩膜向量，实现快速而稳定的预测，解决了现阶段实例分割任务随目标数量增加而预测速度减慢的难题。