insight-face-paddle部署流程

insight-face-paddle部署流程

安装 PaddlePaddle

# for GPU
pip3 install paddlepaddle-gpu

• 注：此处显示paddle安装成功但后续无法正常import

  • 报错信息

    #使用insightfacepaddle时出现Error
    Can not import paddle core while this file exists: /data/User21100768/anaconda3/envs/face/lib/python3.7/site-packages/paddle/fluid/libpaddle.so
    
    ImportError: libcudart.so.10.2: cannot open shared object file: No such file or directory
    

  • 分析

    • 由于libcudart.so是CUDA运行时库的一部分，故推测是Paddle版本与CUDA版本不匹配所导致的

    • 查询服务器的CUDA版本：11.7

      #命令行输入
      nvidia-smi
      

    • https://www.paddlepaddle.org.cn/

      • 从中选择匹配CUDA版本的Paddle，并复制代码到命令行运行（正确配置如下），问题解决
        • 2.4（推荐稳定版)；Linux；pip；CUDA11.7

安装 requirements

pip3 install --upgrade -r requirements.txt -i https://mirror.baidu.com/pypi/simple

安装 InsightFacePaddle

• 方式一：使用pip工具从pypi安装最新版本的 InsightFacePaddle

pip3 install --upgrade insightface-paddle

• 方式二：自己构建并安装

python3 setup.py bdist_wheel
pip3 install dist/*

在命令行中使用 InsightFacePaddle

1. 可以通过以下命令获取参数信息

   insightfacepaddle -h
   

2. 构建索引

   • 如果使用识别功能，则在开始预测之前，必须先构建索引文件，命令如下

   insightfacepaddle --build_index ./demo/friends/index.bin --img_dir ./demo/friends/gallery --label ./demo/friends/gallery/label.txt
   

3. 预测

   • Image(s)

     • 预测命令如下，检测结果图位于路径 ./output 下：

     insightfacepaddle --det --input ./demo/friends/query/friends1.jpg --output ./output
     

   • Video

     • 源码项目文件中没有对应的mp4文件，故从b站解析出相关视频添加到项目文件运行

     insightfacepaddle --det --input ./demo/friends/query/friends.mp4 --output ./output
     

4. 仅识别

   Image(s)

   • 使用下图（某张图片）进行测试

   • 预测命令如下：

     insightfacepaddle --rec --index ./demo/friends/index.bin --input ./demo/friends/query/Rachel.png
     

     检测结果输出在终端中：

     INFO:root:File: Rachel.png, predict label(s): ['Rachel']
     

5. 检测+识别系统串联

   • Image(s)

     • 预测命令如下，检测结果图位于路径 ./output 下

       insightfacepaddle --det --rec --index ./demo/friends/index.bin --input ./demo/friends/query/friends2.jpg --output ./output
       

   • Video

   insightfacepaddle --det --rec --index ./demo/friends/index.bin --input ./demo/friends/query/friends.mp4 --output ./output