手把手教你使用TensorFlow训练出自己的模型

手把手教你使用TensorFlow训练出自己的模型

一、前言
搭建TensorFlow开发环境一直是初学者头疼的问题，为了帮忙初学者快速使用TensorFlow框架训练出自己的模型，作者开发了一款基于TensorFlow 2.3.0的图形化工具AiSuperTool，可以一键加载数据集训练出模型，支持导出h5、tflite格式的模型文件。

二、软件简介
2.1 双击AiSuperTool.exe文件即可打开软件，软件初始化时间大概需15秒，请耐心等待，初始化完成后显示的界面如下：

2.2 点击“加载数据集”按钮，会弹出加载数据集的提示框，如图：

2.3 目前支持加载图片分类的数据集，支持训练图片分类的模型。花卉数据集下载地址：链接：https://pan.baidu.com/s/1ATVF4xHDVTAu-lsj-lC39w
提取码：7rzc
注意：数据集文件夹的应当按照类别存放，比如加载花卉识别数据集时，
data文件夹下是daisy、dandelion、roses、sunflowers、tulips 5个子文件夹，每个子文件下是对应的图片数据。如图：

2.4 加载完图片分类数据集后，点击“开始训练模型”按钮，软件开始训练模型，界面会实时显示训练过程中的loss 损失值和accuracy 准确率以及训练模型的进度。训练时长大概在30分钟内，具体根据电脑配置和数据集的大小有关。训练模型中的界面如下：

> 2.5 训练完成后，会在软件的models文件夹下生成new_model.h5模型文件,这时候可以点击“模型推理预测”按钮来测试模型的好坏，选择要测试的图片后，软件自动加载模型进行推理预测，并将类别和准确率显示在界面上。如图：

2.6 导出模型文件，如果需要将模型文件移植到Android设备中进行推理预测，则需要将h5格式的模型文件转换为tflite格式的模型。软件提供了一键转换功能，点击“导出模型文件”按钮，自动开始进行模型转换，转换完成后在models文件夹生成new_model.tflite模型文件，模型转换过程比较慢，耐心等待即可。

> 2.7 软件下载：https://url83.ctfile.com/f/45573183-861538995-a31a63?p=7526 (访问密码: 7526)