AIAS/7_engine_hub/video_search
2021-12-12 23:09:15 +08:00
..
images video search 2021-12-12 16:35:11 +08:00
video-search Fix Log4j2 漏洞 2021-12-12 23:09:15 +08:00
video-search-ui video search 2021-12-12 16:35:11 +08:00
README.md no message 2021-12-12 21:17:36 +08:00

目录:

http://aias.top/

视频搜索

本例子提供了人像搜索,使用了人脸检测算法,人脸特征提取算法。

Screenshot

主要特性

  • 底层使用特征向量相似度搜索
  • 单台服务器十亿级数据的毫秒级搜索
  • 近实时搜索,支持分布式部署
  • 随时对数据进行插入、删除、搜索、更新等操作

1. 前端部署

1.1 直接运行:

npm run dev

1.2 构建dist安装包

npm run build:prod

1.3 nginx部署运行(mac环境为例)

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

    server {
        listen       8080;
        server_name  localhost;

        location / {
            root   /Users/calvin/Documents/video_search/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     

2. 后端jar部署

2.1 环境要求:

  • 系统JDK 1.8+

  • application.yml
    1). 根据需要编辑图片上传根路径imageRootPath

# 文件存储路径
file:
  mac:
    ...
    imageRootPath: ~/file/image_root/ #图片文件根目录
  linux:
    ....
    imageRootPath: /home/aias/file/image_root/ #图片文件根目录
  windows:
    ...
    imageRootPath: C:\aias\file\image_root\ ##图片文件根目录
    ...

2). 根据需要编辑图片baseurl

image:
  #baseurl是图片的地址前缀
  baseurl: http://127.0.0.1:8089/images/

2.2 运行程序:

# 运行程序

java -jar video-search-0.1.0.jar

3. 后端向量引擎部署docker

3.1 环境要求:

  • 需要安装docker运行环境Mac环境可以使用Docker Desktop

3.2 拉取 向量引擎 镜像(用于计算特征值向量相似度)

sudo docker pull milvusdb/milvus:0.10.0-cpu-d061620-5f3c00

3.3 下载配置文件

vector_engine.zip

3.4 启动 Docker 容器

/Users/calvin/vector_engine为主机路径根据需要修改。conf下为引擎所需的配置文件。

docker run -d --name milvus_cpu_0.10.0 \
-p 19530:19530 \
-p 19121:19121 \
-p 9091:9091 \
-v /Users/calvin/vector_engine/db:/var/lib/milvus/db \
-v /Users/calvin/vector_engine/conf:/var/lib/milvus/conf \
-v /Users/calvin/vector_engine/logs:/var/lib/milvus/logs \
-v /Users/calvin/vector_engine/wal:/var/lib/milvus/wal \
milvusdb/milvus:0.10.0-cpu-d061620-5f3c00

3.5 编辑向量引擎连接配置信息

  • application.yml
  • 根据需要编辑向量引擎连接ip地址127.0.0.1为容器所在的主机ip
##################### 向量引擎 ###############################
search:
  host: 127.0.0.1
  port: 19530
  indexFileSize: 1024 # maximum size (in MB) of each index file
  nprobe: 256
  nlist: 16384
  faceDimension: 512 #dimension of each vector
  faceCollectionName: faces #collection name
  commDimension: 512 #dimension of each vector
  commCollectionName: comm #collection name

4. 打开浏览器

  • 输入地址: http://localhost:8090

  • 视频上传 1). 点击上传按钮上传视频文件.
    测试视频 2). 点击提取人脸特征提取按钮. 等待图片帧解析人脸检测特征提取特征存入向量引擎。通过console可以看到进度信息。 imageRootPath目录下可以看到提取出的图片帧及检测目标图片。

Screenshot

  • 视频搜索 上传图片,点击查询,可以看到返回的图片清单,根据相似度排序。

Screenshot

5. 帮助信息

        String host = "127.0.0.1";
        int port = 19530;
        final String collectionName = "faces"; // collection name

        MilvusClient client = new MilvusGrpcClient();
        // Connect to Milvus server
        ConnectParam connectParam = new ConnectParam.Builder().withHost(host).withPort(port).build();
        try {
            Response connectResponse = client.connect(connectParam);
        } catch (ConnectFailedException e) {
            e.printStackTrace();
        }

        // 检查 collection 是否存在
        HasCollectionResponse hasCollection = hasCollection(client, collectionName);
        if (hasCollection.hasCollection()) {
            dropIndex(client, collectionName);
            dropCollection(client, collectionName);
        }
       ...

官网:

官网链接

Git地址

Github链接
Gitee链接