注册

基于UltraFace的人脸检测在地平线旭日X3派上的部署和测试(Python版本和C++版本)

电脑端的测试环境搭建

如果不想再搭建环境和测试代码bug上浪费更多的时间可以直接获取本人的测试虚拟机,所有的测试代码、虚拟环境和板端测试工程以全部打包到了虚拟机,需要的可以通过网盘获取:

代码和虚拟机百度网盘链接:

链接:https://pan.baidu.com/s/15Np6Hn4_WuFDqcQ40aCe7A 提取码:j1ez 

搭建开发环境

#创建虚拟环境
conda create -n UltraFace python=3.6
conda activate UltraFace

pip install opencv-python==3.4.6.27  -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install onnx==1.8.1   -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install future   -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install protobuf==4.21.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install onnxruntime -i https://pypi.tuna.tsinghua.edu.cn/simple

pth模型测试

#激活虚拟环境
conda activate UltraFace
#测试路径
cd /home/kkx3/wyj/UltraFace/Ultra-Light-Fast-Generic-Face-Detector-1MB-master
#测试指令
python run_video_face_detect_640.py  --test_device cpu  

onnx模型测试

#激活虚拟环境
conda activate UltraFace
#测试路径
cd /home/kkx3/wyj/UltraFace/Ultra-Light-Fast-Generic-Face-Detector-1MB-master
#测试指令
python run_video_face_detect_onnx_640.py  --test_device cpu  

pth模型转onnx模型

#测试路径
cd /home/kkx3/wyj/UltraFace/Ultra-Light-Fast-Generic-Face-Detector-1MB-master
#模型转换指令
python convert_to_onnx_640.py

onnx模型转换bin模型

#激活虚拟环境
conda activate x3model
#测试路径
cd /home/kkx3/wyj/x3/TransModel/horizon_model_convert_sample/04_detection/09_Face_RFB/mapper
#模型转换
bash 01_check.sh
bash 02_preprocess.sh
bash 03_build.sh

需要修改配置文件

1、01_check.sh,修改模型路径

2、02_preprocess.sh修改测试图片路径

3、RFB_config.yaml修改配置文件

4、03_build.sh,修改模型路径

5、输出的模型路径

6、preprocess.py修改分辨率

python版本板端测试

链接开发板,复制模型问文件到板端

#测试路径
cd /app/pydev_demo/14_FaceRFB_sample
#测试指令
./test_yolov5.py

修改分辨率和模型

测试图片

测试结果

c++版本板端测试

复制模型到板端

#测试路径
cd /app/cdev_demo/bpu_FaceRFB
#编译
cd /app/cdev_demo/bpu_FaceRFB/src
make clean && make 
#运行
cd /app/cdev_demo/bpu_FaceRFB/src/bin
./sample -f ./RFB_480x640_nv12.bin -m 0

测试结果视频

-=||=-赞 (11)
未经允许不得转载:金书世界 » 基于UltraFace的人脸检测在地平线旭日X3派上的部署和测试(Python版本和C++版本)

相关文章

评论 (0)