星瞳AI云服务
快速上手

制作分类模型

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

介绍

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

前提条件

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

准备图片

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

新建项目

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

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

上传图片

  1. 进入项目后默认就是「数据集」工作区,也可以点击顶栏的「数据集」

  1. 点击左栏顶部的「上传」

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

创建标签

  1. 点击右侧标签面板右上角的「+」
  2. 输入标签名称,每行一个,这里输入 cat 和 dog,然后点击「创建」即可

标注

  1. 在左栏点击一张图片
  2. 在右侧标签面板点击要标注的标签名字(快捷键:标签前面的序号数字键)
  3. 点击「保存」即可(快捷键:回车),会自动跳到下一张图片
  4. 将所有的图片按照上面的步骤标注完。这是最耗费时间的步骤,也是最关键的步骤。

训练

  1. 点击顶栏的「训练」,再点击左侧训练列表右上角的「+」

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

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

下载模型

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

On this page