AI训练平台提供分类模型训练能力。并以REST API形式为上层应用提供接口。 当前版包含功能如下: -分类模型训练 (imagenet数据集预训练的resnet50模型) -模型训练可视化 -图片分类推理 -图片特征提取(512维特征) -图片 1:1 比对
输入地址: http://localhost:8080
压缩包内需包含2个目录(名字需严格一致):
-TRAIN :包含训练数据,每个文件夹对应一个分类(每个分类图片数量尽量保持平衡)
-VALIDATION :包含验证数据,每个文件夹对应一个分类
-选择zip文件并上传 -点击训练按钮开始训练
图片特征提取使用的是新训练的模型。特征来自模型的特征提取层。