点击返回网站首页

 

AI 训练平台

AI训练平台提供分类模型训练能力。并以REST API形式为上层应用提供接口。 当前版包含功能如下: -分类模型训练 (imagenet数据集预训练的resnet50模型) -模型训练可视化 -图片分类推理 -图片特征提取(512维特征) -图片 1:1 比对

打开浏览器

输入地址: http://localhost:8080

1. 训练数据准备-ZIP格式压缩包:

压缩包内需包含2个目录(名字需严格一致):
-TRAIN :包含训练数据,每个文件夹对应一个分类(每个分类图片数量尽量保持平衡)
-VALIDATION :包含验证数据,每个文件夹对应一个分类

2. 上传数据并开始训练:

-选择zip文件并上传 -点击训练按钮开始训练

3. 查看训练过程:

4. 图片分类测试:

5. 特征提取测试:

图片特征提取使用的是新训练的模型。特征来自模型的特征提取层。

6. 图片比对测试:

项目源代码下载:

Github链接

Gitee链接

 

点击返回网站首页