查看原文
其他

视觉提示新突破 | IDEA研究院发布T-Rex模型,解锁 「一图胜千言」的Prompting新体验

IDEA研究院 IDEA数字经济研究院
2024-08-22
继火爆出圈的Grounded SAM之后,IDEA研究院团队携重磅新作归来:视觉提示(Visual Prompt)模型T-Rex,以图识图,开箱即用,开启开集检测新天地!



“拉框、检测、完成!”
在刚刚结束的2023 IDEA大会上,IDEA研究院创院理事长、美国国家工程院外籍院士沈向洋展示了基于视觉提示的目标检测新体验,并发布了全新视觉提示模型T-Rex的模型实验室(playground), Interactive Visual Prompt(iVP)掀起现场一波试玩小高潮。(前往 https://deepdataspace.com/playground/ivp 亲自试玩)




在iVP上,用户可以亲自解锁“一图胜千言”的 Prompting:在图片上标记感兴趣的对象,向模型提供视觉示例,模型随即检测出目标图片中与之相似的所有实例。整套流程交互便捷,只需几步操作就可轻松完成。


操作演示


IDEA研究院4月份发布的Grounded SAM (Grounding DINO + SAM) 曾在Github上火爆出圈,至今已狂揽11K星。有别于只支持文字提示的Grounded SAM,此次发布的T-Rex模型提供着重打造强交互的视觉提示功能。 T-Rex具备极强的开箱即用特性,无需重新训练或微调,即可检测模型在训练阶段从未见过的物体。该模型不仅可应用于包括计数在内的所有检测类任务,还为智能交互标注领域提供新的解决方案。
T-Rex应用潜力广泛
团队透露,研发视觉提示技术是源自对真实场景中痛点的观察。有合作方希望利用视觉模型对卡车上的货物数量进行统计。然而,仅通过文字提示,模型无法单独识别出每一个货物。其原因是工业场景中的物体在日常生活中较为罕见,难以用语言描述。在此情况下,视觉提示显然是更高效的方法。与此同时,直观的视觉反馈与强交互性,也有助于提升检测的效率与精准度。 


基于对实际使用需求的洞察,团队将T-Rex设计成可以接受多个视觉提示的模型,且具备跨图提示能力。除了最基本的单轮提示模式,目前还支持以下三种进阶模式。


· 多轮正例模式:适用于视觉提示不够精准造成漏检的场景

· 正例+负例模式: 适用于视觉提示带有二义性造成误检的场景

· 跨图模式:适用于通过单张参考图提示检测他图的场景


在同期发布的技术报告中,团队总结了T-Rex模型的四大特性:

T-REX模型四大特性


开放集:不受预定义类别限制,具有检测一切物体的能力

视觉提示:利用视觉示例指定检测目标,克服罕见、复杂物体难以用文字充分表达的问题,提高提示效率

直观的视觉反馈:提供边界框等直观视觉反馈,帮助用户高效评估检测结果

交互性:用户可通过便捷交互,积极参与检测过程,对模型结果进行纠错


T-REX模型由图像编码器、提示编码器、框解码器组成
研究团队指出,在目标检测场景中,视觉提示的加入能够补足文本提示的部分缺陷。未来,两者的结合将进一步释放CV技术在更多垂直领域的落地潜能。

iVP 模型实验室:https://deepdataspace.com/playground/ivp(可点击阅读原文跳转)

报告链接: https://arxiv.org/abs/2311.13596 

Github链接:trex-counting.github.io


本项工作来自IDEA研究院计算机视觉与机器人研究中心(CVR)。该团队此前开源的目标检测模型DINO是首个在COCO目标检测上取得榜单第一的DETR类模型。在Github上大火的零样本检测器Grounding DINO与能够检测、分割一切的Grounded SAM,同样为该团队作品。



彩蛋

T-Rex,来自于“Tyrannosaurus Rex”的缩写,中文名为霸王龙,也是CVR团队DINO恐龙系列工作的延伸。一个关于霸王龙的冷知识:经过科学家的研究,霸王龙是各种恐龙中视力最好的,是人的视力的13倍,甚至比鹰的视力还好(鹰的视力只是人的3.6倍)。


T-Rex模型的Logo是一只刚破壳而出的可爱小霸王龙,寓意着T-Rex模型刚刚诞生,但有着巨大的发展潜力,有朝一日将进化得更加强大。



关于 IDEA研究院CVR


计算机视觉与机器人研究中心(CVR,Computer Vision and Robotics)立足于计算机视觉和机器人方向的基础研究,专注于大规模视觉表示学习、物体检测与识别、智能控制等问题,通过核心技术的研究和突破,打造世界领先的机器视觉和智能机器人技术。同时,CVR力图通过对技术和平台的开源来造福产业和基础科研。目前,CVR的开源方向主要包括detrex物体检测框架和DeepData Space数据平台。



继续滑动看下一个
IDEA数字经济研究院
向上滑动看下一个

您可能也对以下帖子感兴趣

文章有问题?点此查看未经处理的缓存