Video2TagVideo2Tag

泥嚎!Video2Tag

DeDeon 15 days ago

使用Video2Tag和MaixHub创建和训练AI数据集的流程指南

本文档将引导您完成从视频素材到可在MaixHub上使用的AI训练数据集的完整流程。我们将使用Video2Tag工具来处理视频和标注对象,然后将生成的数据集导入MaixHub进行模型训练。

一、拍摄高质量的视频数据

高质量的原始数据是训练出优秀AI模型的基础。在开始拍摄用于制作数据集的MP4视频文件之前,强烈建议您参考以下博文,了解如何拍摄更好的数据集和视频:

这篇博文将为您提供关于光照、角度、多样性等方面的实用建议,以确保您的视频素材更适合后续的AI模型训练。

二、使用Video2Tag处理视频并创建数据集

Video2Tag工具可以帮助您高效地从视频中提取图像并进行标注。

  1. 上传视频:

    • 将您拍摄好的MP4视频文件拖拽到Video2Tag的上传区域。
    • 耐心等待视频上传成功。
  2. 标注对象:

    • 视频上传完成,显示图片后,您可以在视频帧中开始标注。
    • 在您希望模型识别的物体上精确地画出边界框。
  3. 输入标签:

    • 在界面的右侧(或指定区域)为每个框选的物体输入相应的标签名称。
    • 在界面的顶部去选择处理视频的方法,对于分辨率,如果是K210设备,可以选择480P,如果是K230或是MaixCam可以选择720P,如果是电脑运行的YOLO,可以根据自己的摄像头参数选择,分辨率是向下采样,类似我输入480P的视频,即使选择了1080P,最终导出的数据集依然是480P;
    • 其中的帧率是为了更好的减少数据集重复的部分,类似我们拍摄的视频一般都是30fps,不过帧与帧之间都差不多,抽帧可以尽可能减少重复部分,推荐10fps,是追踪模型支持+减少重复比较平衡的数值。
  4. 确认与处理:

    • 等待系统处理视频,获取视频属性。
    • 显示处理完成就可以点击打标签啦。
    • 等待处理完成。
  5. 预览与下载:

    • 处理完成后,您可以预览标注效果,检查是否有误。
    • 点击“下载VOC数据集”按钮。下载的文件通常是一个压缩包,这个数据集已经为在MaixHub上使用做好了准备。如果是自己训练YOLO,请选择YOLO格式

三、在MaixHub上导入数据集并开始训练

MaixHub是一个AI模型训练平台,您可以使用前面步骤中创建的数据集在这里训练自己的模型。

  1. 访问MaixHub:

    • 您可以直接在平台的底部找到MaixHub的入口。 image
  2. 开始训练流程:

    • 在MaixHub中选择“开始训练”选项。 image
  3. 管理数据集:

    • 您可以选择“新建数据集”来创建一个全新的数据集项目。 image
    • 或者,如果您之前已有数据集,也可以选择“编辑之前的数据集”。
  4. 导入数据集:

    • 选择“导入数据集”功能。 image
    • 在文件选择对话框中,选择您刚刚从Video2Tag下载的那个数据集压缩包文件。 image
  5. 查看数据集:

    • 导入成功后,您应该能看到数据集已经包含了从视频中提取的、涵盖各个角度的图像。
    • 我们之前完成的数据集标签也会一并展示出来,可供模型训练使用。

通过以上步骤,您就可以顺利地利用视频素材,通过Video2Tag工具制作出带有标注的数据集,并将其导入MaixHub平台进行后续的AI模型训练了。祝您训练顺利!