很高兴地宣布,Roboflow与CVAT(计算机视觉标注工具)的集成将为图像标注过程带来革命性的改变。通过这一集成,可以将Roboflow Universe中超过11,000个公开模型直接加载到CVAT中,从而加速标注工作。对于前1000张图片,这一服务是完全免费的。
在本指南中,将详细介绍如何将Roboflow模型加载到CVAT中。首先,需要拥有一个Roboflow账户。如果还没有,可以在Roboflow的中注册。同样,如果还没有CVAT账户,也可以在上创建一个。
步骤1:在Universe中选择模型
可以在CVAT中使用Universe上的公共模型。要探索Roboflow Universe,请前往首页并根据正在寻找的对象类别进行搜索。点击不同的数据集,查看哪些数据集包含了需要的类别。如果想要在CVAT中使用模型,那么数据集需要包含一个训练好的模型。可以通过点击“模型”侧边栏链接来检查是否有可用的模型。
在本指南中,将使用一个在Roboflow Universe上公开的模型,该模型能够在各种环境中识别人和梯子。将使用这个模型来帮助标注更多的图片,以便进一步改进模型。
步骤2:将模型添加到CVAT
要将Roboflow模型加载到CVAT中,首先点击CVAT导航栏中的“模型”链接。在另一个标签页中,转到想要使用的模型的Roboflow Universe页面,然后点击侧边栏中的“模型”。复制页面的URL。然后,将想要使用的Roboflow模型的链接粘贴到CVAT中:
CVAT会自动将提供者识别为“Roboflow”。需要API密钥来使用该模型。要获取此密钥,请滚动到标题为“Infer on Local and Hosted Images”的部分,并复制Universe页面上Python代码片段中的API密钥。复制“api_key”后的文本并粘贴到CVAT中,然后点击“提交”。
返回CVAT的“模型”页面。将看到添加的模型已列在那里:
现在已经将模型加载到CVAT中,可以开始在标注中使用它了。
步骤3:标注图像
在CVAT标注界面中,将鼠标悬停在侧边栏的魔杖图标上。然后,点击“检测器”。这将允许选择要使用的模型来帮助标注图像。选择在上一步中添加到CVAT的模型:
接下来,需要将Roboflow模型中的类别名称映射到想要在CVAT中使用的标签。点击第一个下拉菜单并选择Roboflow模型中的类别名称。然后,点击第二个下拉菜单并添加相应的类别用于CVAT标注:
在这个例子中,将Roboflow的“梯子”预测映射到了标注环境中的“梯子”标签,并将“人”映射到了“人”标签。当Roboflow模型识别出一个人时,模型的预测将在浏览器中以“人”标签绘制出来。点击“标注”以加载模型并开始使用模型辅助标注。
这是在示例图像上运行模型的结果:
模型成功识别了人和梯子,并分配了正确的标签。可以将此模型应用于更多图像,以加快标注速度,消除手动绘制每个感兴趣对象的边界框的需要。
结论
通过结合使用Roboflow和CVAT,可以比以往更快地标注存储在CVAT中的图像。可以使用Roboflow Universe上成千上万的公开训练模型作为标注助手。