星瞳AI云服务
快速上手

制作分类模型

此教程介绍了如何制作分类模型。

介绍

分类是常见的图像识别任务,它可以将图片分类,比如分类一个图片是猫还是狗。

前提条件

注册帐号,并登录。参考:账号管理

准备图片

  1. 图片是用来训练模型的,图片越多,模型的准确度就越高。
  2. 通常每个类别需要 1000 张图片,可以达到比较好的效果。
  3. 图片尽量覆盖所有可能的环境。

新建项目

  1. 在首页,点击 「创建项目」 按钮。

  1. 输入项目名称并选择「图像分类」,再点击确认即可。

上传图片

  1. 点击左侧的「数据集」即可看到数据集列表

  1. 点击「上传」

  1. 点击「选择文件」,选择图片,可多选,点击确认即可

创建标签

  1. 点击左侧侧边栏中的「标签」进入标签列表
  2. 点击右上角「创建标签」

  1. 输入标签名称,比如 cat,然后点击「确认」即可,这里创建一个 cat 标签一个 dog 标签

标注

  1. 点击左侧侧边栏中的「标注」。
  2. 点击要标注的标签名字。

  1. 点击「保存」即可(快捷键:空格键)

  1. 将所有的图片按照上面的步骤标注完。这是最耗费时间的步骤,也是最关键的步骤。

训练

  1. 在训练页面,点击「创建训练」

  1. 输入训练任务名称
  2. 选择设备和训练的模型
  3. 点击「开始训练」即可,这可能需要一些时间,训练的时间取决于数据集的大小和训练轮数。

关于参数和结果,参考:训练管理

下载模型

在「部署」> 「下载模型文件」即可下载训练完的模型,下载后的文件可用于上传到摄像头。

On this page