1. 磐创智能首页
  2. TensorFlowNews

多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!

点击上方“磐创AI”,选择“置顶公众号”

精品文章,第一时间送达

多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!

转载自:图灵TOPIA,未经允许不得二次转载


图像标注速度提升10倍!


这是多伦多大学与英伟达联合公布的一项最新研究:Curve-GCN的应用结果。


Curve-GCN是一种高效交互式图像标注方法,其性能优于Polygon-RNN++。在自动模式下运行时间为29.3ms,在交互模式下运行时间为2.6ms,比Polygon-RNN ++分别快10倍和100倍。



数据标注是人工智能产业的基础,在机器的世界里,图像与语音、视频等一样,是数据的一个种类。


对象实例分割是在图像中概括给定类的所有对象的问题,这一任务在过去几年受到了越来越多的关注,传统标记工具通常需要人工先在图片上点击光标描记物体边缘来进行标记。


然而,手动跟踪对象边界是一个费力的过程,每个对象大概需要30-60秒的时间。


多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!


为了缓解这个问题,已经提出了许多交互式图像分割技术,其通过重要因素加速注释。但是交互式分割方法大多是逐像素的(比如DEXTR),在颜色均匀的区域很难控制,所以最坏的情况下仍然需要很多点击。


Polygon-RNN将humans-in-the-loop(人机回圈)过程进行构架,在此过程中模型按顺序预测多边形的顶点。通过纠正错误的顶点,注释器可以在发生错误时进行干预。该模型通过调整校正来继续其预测。Polygon-RNN显示在人类协议水平上产生注释,每个对象实例只需点击几下。这里最糟糕的情况是多边形顶点的数量,大多数对象的范围最多为30-40个点。


然而,模型的重复性将可扩展性限制为更复杂的形状,导致更难的训练和更长的推理。此外,期望注释器按顺序纠正错误,这在实践中通常是具有挑战性的。


多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!

多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!


最新研究成果中,研究人员将对象注释框架化为回归问题,其中所有顶点的位置被同时预测。


在Curve-GCN中,注释器会选择一个对象,然后选择多边形或样条轮廓。


多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!

Curve-GCN自动地勾勒出对象的轮廓


多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!

Curve-GCN允许交互式更正,并且可以自动重新预测多边形/样条


与Polygon-RNN + +相比:


  • Curve-GCN具有多边形或样条曲线参数

  • Curve-GCN可同时预测控制点(更快)


多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!

多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!

初始化预测


多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!

人工校正


多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!

模型校正


多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!

模型是在CityScapes数据集上训练的


多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!

自动模式下的比较。从左到右:ground-truth, Polygon-GCN, Spline-GCN, ps – deeplab


多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!

跨域自动模式。(上)cityscaps训练模型的开箱即用输出,(下)使用来自新领域的10%的数据进行微调。

多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!

多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!

将Polygon和Spline-GCN与Polygon-RNN ++和PSP-DeepLab进行比较


模型在最先进的基础上进行了改进,速度显著加快,允许只具有局部效果的交互式更正,从而为注释器提供了更多的控制。这将导致更好的整体注释策略。


代码:

https://github.com/fidler-lab/curve-gcn


论文地址:

https://arxiv.org/abs/1903.06874


论文摘要


通过边界跟踪来手动标记对象是一个繁重的过程。在Polygon-RNN ++中,作者提出了Polygon-RNN,它使用CNN-RNN架构以循环方式产生多边形注释,允许通过humans-in-the-loop(人机回圈)的方式进行交互式校正。


我们提出了一个新的框架,通过使用图形卷积网络(GCN)同时预测所有顶点,减轻了Polygon-RNN的时序性。我们的模型是端到端训练的。它支持多边形或样条对对象进行标注,从而提高了基于线和曲线对象的标注效率。结果表明,在自动模式下,Curve-GCN的性能优于现有的所有方法,包括功能强大的PSP-DeepLab,并且在交互模式下,Curve-GCN的效率明显高于Polygon-RNN++。我们的模型在自动模式下运行29.3ms,在交互模式下运行2.6ms,比Polygon-RNN ++分别快10倍和100倍。


多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!


多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!


你也许还想

   GitHub超全机器学习工程师成长路线图,开源两日收获3700+Star!

   使用Python+OpenCV进行图像处理(二)| 视觉入门

   上Github,北大、清华、浙大、中科大4大名校课程在线学,加星总数超1.8万

欢迎扫码关注:

多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!


觉得赞你就点在看,多谢大佬多伦多大学&NVIDIA最新成果:图像标注速度提升10倍!

磐创AI:http://www.panchuangai.com/ 智能客服:http://www.panchuangai.com/ TensorFlow:http://panchuang.net 推荐关注公众号:磐创AI

原创文章,作者:fendouai,如若转载,请注明出处:https://panchuang.net/2019/04/20/098ad5d20f/

发表评论

电子邮件地址不会被公开。

联系我们

400-800-8888

在线咨询:点击这里给我发消息

邮件:admin@example.com

工作时间:周一至周五,9:30-18:30,节假日休息