Prodigy与PyTorch实现图像标注工作流

0 阅读1分钟

Prodigy 是一款现代化的标注工具,用于收集机器学习模型的训练数据。本文将展示如何使用 Prodigy 在 Python 中编写完全自定义的标注工作流,如何接入自己的机器学习模型,以及如何针对特定用例混合搭配不同界面。具体操作包括:创建一个图像描述数据集,使用基于 PyTorch 实现的图像描述模型来生成建议描述,并进行错误分析,以找出模型的正确预测和需要改进之处。

分步指南

  1. 01:59 – 创建手动图像标注的 recipe
  2. 13:26 – 接入基于 PyTorch 实现的图像描述模型并修正其输出
  3. 19:41 – 添加回调函数以统计修改过的描述并输出结果
  4. 26:33 – 添加用于错误分析和审核标注的工作流

相关资源