图像识别AI程序（数据集管理/训练/预测）

出于对Python中AI智能识别程序的兴趣，对照AI智能识别程序应该具有的主要功能：数据集管理（加载/移除/重命名）、模型训练（可实时监测训练进度、最终训练参数输出）、模型预测，使用Python/Keras编制了AIBox工具，可供大家使用/参考。

图1 AiBox主界面

一、数据集管理

（1）数据集添加

通过“导入数据集”工具栏按钮，打开数据集导入窗口，选择数据集目录（也可以选择数据集目录所在的上级目录，以一次加载所有数据集）。

图2 数据集加载

加载后的数据集显示如下所示：

图3 数据集显示

（2）数据集移除

可将不需要的数据集目录，在左侧进行选中，并点击工具栏“删除数据集”按钮将其移除。

图4 数据集移除

二、模型训练

首先，点击“训练模型”右侧的向下展开按钮，以在训练过程中监测实时进展和相关数据。

然后，点击“训练模型”即可开始训练。

图5 模型训练过程展示

三、模型预测

首先点击左下方“模型预测”按钮，进入模型预测窗口。

第一步：载入预测所用模型；

第二步：载入待预测图片；

第三步：点击“预测”，进行图片概率预测。

图6 图形预测窗口

感兴趣的朋友可以私聊我获取程序源码，目前测试中尚存在一些小问题，暂无时间解决，望见谅。

最近会先进行打包，将该程序的发布版共享出来。

联系QQ：272464982