概览硬件软件云平台
OAK 应用
DepthAI
AI
软件栈
故障排除

本页目录

  • 概述
  • 使用方法
  • 数据集选择
  • 训练
  • 后续步骤

Roboflow Universe

Supported on:RVC2RVC4

概述

Roboflow 是一个综合性平台，旨在简化构建、训练和部署计算机视觉模型的过程。其关键产品之一是 Roboflow Universe，这是一个社区驱动的平台，用户可以在其中分享和探索各种应用场景下的标注计算机视觉数据集。在此集成中，我们重点介绍如何使用 Roboflow Universe 的预标注数据集，并将其应用于您自己的自定义模型开发流程。我们在 Roboflow Universe 与 LuxonisTrain 框架之间构建了无缝集成。本指南将带您了解主要步骤。如需更详细的实践示例，请查看我们的 Colab 教程。

使用方法

数据集选择

首先访问 Roboflow Universe，搜索与您的任务相关的数据集。在搜索栏中输入相关关键词，即可获得大量数据集供您探索。选择数据集时，请考虑以下几点：
  • 标注类型必须与预测目标匹配：对于目标检测模型，数据集必须包含边界框标注。对于实例分割，标注除了边界框外，还必须包含每个对象的遮罩。其他类型的任务以此类推。
  • 数据集质量很重要：确保标注准确、图像内容多样且数据集足够大。我们通常建议至少包含 500–1000 张图像 以获得稳健性能——但更多数据效果更好。
  • 图像领域应与部署环境匹配：训练图像应尽可能接近模型在生产环境中看到的场景。例如，如果您的工厂只使用蓝色 PCB，但数据集中全是绿色 PCB，则模型可能难以泛化。您可以尝试使用颜色增强来帮助解决领域偏移，或搜索更适合您部署环境的数据集。

训练

选择数据集后，您可以在 LuxonisTrain 配置文件 中引用它。需要在 loader 的 params 部分设置关键参数 dataset_dir。我们为 Roboflow 数据集使用特殊的 URI 格式：
Markdown
1roboflow://<团队名称>/<数据集名称>/<数据集版本>/coco
您可以直接从数据集 URL 中提取这些组成部分。例如，对于以下数据集：
Markdown
1https://universe.roboflow.com/learn-uzoux/pcb-defect-0i1a7
对应的 dataset_dir 为：
Markdown
1roboflow://learn-uzoux/pcb-defect-0i1a7/2/coco
正确指定数据集后，即可开始训练。您可以通过我们的 Colab 教程 了解后续步骤——训练、验证和部署。

后续步骤

如果初始训练后模型性能不理想，请考虑调整配置——例如，增加训练轮数、使用更大的架构或调整超参数。但通常，最佳改进来自于添加更多高质量数据。Roboflow 使您可以轻松地派生现有数据集、合并新图像，并在需要时重新标注。如果数据集是私有的，请记得在训练前重新导出您的 API 密钥（如上所述）。