你的位置:首页 > 信息动态 > 新闻中心
信息动态
联系我们

4步操作,完成Boxlnst模型复现

2021-11-2 16:44:11

一、模型介绍

《BoxInst: High-Performance Instance Segmentation with Box Annotations》于2020年11月发表,出自沈春华组,提出了一个实现 mask 级实例分割的高性能方法,训练时只需使用边框标注。本文中,作者只用到了一个简单的设计就可以实现显著的性能提升,在COCO数据集上 mask AP 由 21.1 % 21.1\%21.1%大幅提升到了31.6 % 31.6\%31.6%。本文核心思想重新设计实例分割中的 mask 学习损失函数,无需修改分割网络本身。新的损失函数无需 mask 标注,就可监督 mask 的训练。它是通过以下两个损失项做到的:1) 一个最小化 ground-truth 边框映射和预测 mask 差异的替代项;2) 一对损失使得颜色相近的相邻像素点类别相同。 

接下来我们一起看看如何快速进行论文复现吧!

二、创建实例

首先打开极链AI云的官网,选择模型创建实例:

模型列链AI云平台

由于平台已经配置了Boxinst的模型库,所以能够一键创建实例,轻松又便捷

三、论文复现

1、在/root/AdelaiDet/datasets/路径下,新建名为coco的文件夹

2、在根目录终端中,输入数据集准备代码

3、在/root/AdelaiDet文件夹下的终端中输入训练代码

训练效果如下

3、在目录/root/AdelaiDet的终端下,输入下列代码,使用预训练权重文件

到这里,我们就已经完成了Boxlnst的整个复现成果~具体的模型简介和代码可以在模型详情页查看

为了使复现更加方便高效,我们为每一个模型都配置了专属的模型详情页。

模型详情页详细展现了模型的使用方法(包含训练和推理的使用方式测试案例)在证明模型配置的完整性和方便用户直接测试模型效果上,为用户进一步开发和训练模型提供便利

极链AI云平台现在更新了大量模型和数据库!省去了繁琐冗长的上传时间,让论文复现一步到位!快去官网试试吧

ps.在平台没有找到模型的小伙伴可以参加我们的模型复现活动,在学习同时更有云币券作为奖励,快戳小助手咨询详情吧