SynergyNet(头部姿态估计 Head Pose Estimation)复现 demo测试

0 相关资料

SynergyNet(github):https://github.com/choyingw/SynergyNet

b站:https://www.bilibili.com/video/BV1Ph4y1k7LK/

1 环境搭建

我用的AutoDL平台搭建

选择镜像
PyTorch 1.9.0
Python 3.8(ubuntu18.04)
Cuda 11.1
在这里插入图片描述

安装其它包:

pip install opencv-python
pip install scipy
pip install matplotlib
pip install cython 

2 安装 SynergyNet

git clone https://github.com/choyingw/SynergyNet
cd SynergyNet 

备注:如果无法下载,就到github中下载压缩包,然后上传解压安装

或者使用码云下载:

git clone https://gitee.com/YFwinston/SynergyNet
cd SynergyNet 

3 下载相关文件

需要下载如下文件:
在这里插入图片描述

如果你无法使用google drive,我把这三个文件都上传到了csdn中,如下:

扫描二维码关注公众号,回复: 16438370 查看本文章

best.pth.tar: https://download.csdn.net/download/WhiffeYF/88175489
aflw2000-data.zip:https://download.csdn.net/download/WhiffeYF/88175374
3dmm-data.zip:https://download.csdn.net/download/WhiffeYF/88175101

将上面的aflw2000-data.zip,3dmm-data.zip上传到SynergyNet文件夹下,然后解压

在这里插入图片描述
将best.pth.tar上传到SynergyNet/pretrained文件夹下
在这里插入图片描述

4 编译

在SynergyNet目录下编译

cd Sim3DR

./build_sim3dr.sh

在这里插入图片描述

cd ../FaceBoxes

./build_cpu_nms.sh

cd ..

在这里插入图片描述

5 测试

在在SynergyNet目录下输入

python singleImage.py -f img

在这里插入图片描述

在这里插入图片描述

测试自己的图片

python singleImage.py -f img/1.png

在这里插入图片描述
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/WhiffeYF/article/details/132113788