一种实例分割模型训练方法、装置及可读介质

    公开(公告)号:CN115601546A

    公开(公告)日:2023-01-13

    申请号:CN202211324514.3

    申请日:2022-10-27

    Abstract: 本发明公开了一种实例分割模型训练方法、装置及可读介质,通过采用第一训练集分别训练主体模型和辅助模型,得到经训练的主体模型和经训练的辅助模型,其中,主体模型为实例分割模型,辅助模型为逐像素分类模型;将待测图像分别输入经训练的主体模型和经训练的辅助模型,得到第一预测结果和第二预测结果,分别获取第一预测结果和第二预测结果的第一掩膜图和第二掩膜图;根据第一掩膜图和第二掩膜图确定难例样本,难例样本为因置信度低而未被经训练的主体模型识别出目标的待测图像,将难例样本与其第二预测结果对应的标签保存为第二训练集;采用第二训练集对经训练的主体模型继续进行训练,得到训练好的主体模型,能够有效提高模型泛化性和准确率。

Patent Agency Ranking