昇腾社区首页
中文
注册

模型适配工具说明

工具简介

模型适配工具是一款集成数据集管理、模型训练、模型打包为一体的开发者工具套件,预置了4个典型模型,分别用于目标检测场景(YOLO V5模型)、图像分类场景(Mobilenet V3模型)、图像分割场景(Unet ++模型)、关键点检测场景(Alphapose模型支持用户使用PCCPU算力训练模型并对模型进行打包,拷贝到开发者套件进行推理。

图1 工具架构

工具优点

  • 工具预置了典型模型,用户可以根据任务场景选择使用
  • 使用工具完成训练创建,全UI界面操作,技术门槛低,适合新手初步了解AI。
  • 使用PC的CPU算力进行训练,降低硬件成本。