HyperAIHyperAI

Command Palette

Search for a command to run...

LVIS挑战赛2020年一等奖解决方案:一个良好的边界框并不能保证一个良好的掩码

Jingru Tan Gang Zhang Hanming Deng Changbao Wang Lewei Lu Quanquan Li Jifeng Dai

摘要

本文介绍了lvisTraveler团队在LVIS Challenge 2020中的解决方案。在本工作中,主要考虑了LVIS数据集的两个关键特性:长尾分布问题以及高质量实例分割掩码。为此,团队采用两阶段训练流程:在第一阶段,引入EQL(Efficient Quality Learning)与自训练(self-training)策略,以学习更具泛化能力的特征表示;在第二阶段,采用Balanced GroupSoftmax优化分类器,并提出一种新颖的候选框分配策略以及一种新的平衡掩码损失函数(balanced mask loss),以提升掩码头的预测精度。最终,该方法在LVIS v1.0验证集和测试开发集上分别取得了41.5和41.2的AP(Average Precision)成绩,显著优于基于X101-FPN-MaskRCNN的基线模型。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供