【新手基础教程】训练自己的分类和检测模型

训练自己的分类和检测模型

Maixhub 模型训练

Maixhub 提供模型训练功能和模型分享功能， 只需要准备好需要训练的数据集， 不需要搭建训练环境和代码， 上传训练数据即可快速训练出模型

目前 Maixhub 支持分类模型 和 目标检测模型的训练。

你需要做的：

• 确定目标， 是分类还是检测
• 根据使用说明制作符合要求的数据集
• 上传数据集等待云端自动训练
• 训练完成， 结果会通过邮件进行通知， 不管是成功还是失败，都会有邮件通知，里面有详细的任务信息和结果文件

Maixhub 使用说明参考 Maixhub 训练使用说明

本地模型训练

本地模型训练使用 sipeed/maix_train 这份代码进行， 使用了 Tensorflow 作为训练框架

主要支持：

• 物体分类模型（使用 Mobilenet V1）： 只识别图片中的物体是什么
• 物体检测模型（使用 YOLO V2）： 找到图片中认识的图体，并同时找到其坐标和大小

1. 系统环境

首先，需要一台有 Linux 系统的电脑 如果你的主力系统是 Windows， 你可以用以下系统环境：

• 使用虚拟机,