人工智能系列：AI 可视化训练平台

本文链接：https://blog.csdn.net/weixin_39355136/article/details/121017585

AI 训练平台

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

前端部署

nginx部署运行：

cd /usr/local/etc/nginx/
vi /usr/local/etc/nginx/nginx.conf
# 编辑nginx.conf

    server {
        listen       8080;
        server_name  localhost;

        location / {
            root   /Users/calvin/platform/dist/;
            index  index.html index.htm;
        }
     ......
     
# 重新加载配置：
sudo nginx -s reload 

# 部署应用后，重启：
cd /usr/local/Cellar/nginx/1.19.6/bin

# 快速停止
sudo nginx -s stop

# 启动
sudo nginx

后端

# 运行程序

java -jar aais-platform-train-0.1.0.jar

打开浏览器

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

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

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

Screenshot

-320张车辆图片测试数据下载

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

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

3. 查看训练过程:

Screenshot

4. 图片分类测试:

Screenshot

5. 特征提取测试:

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

6. 图片比对测试:

Screenshot

7. API文档：

http://127.0.0.1:8089/swagger-ui.html
Screenshot

官网：

http://www.aias.top/

Github地址：

https://github.com/mymagicpower/AIAS
https://gitee.com/mymagicpower/AIAS