新闻  |   论坛  |   博客  |   在线研讨会
代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)
CV研究院 | 2020-11-25 19:39:13    阅读:1998   发布文章

1.png2.jpg

3.png4.png

—— 01 ——

5.png

—— 02 ——

6.png

部分代码如下:

7.png8.png

9.png

部分代码如下:

10.png

—— 03 ——

模型的调用和显示

(1)根据点坐标绘制标注图形:

部分代码如下:

11.png12.png

部分代码如下:

13.png

效果如下图所示:

14.png

Demo

# install requirements

First install Anaconda3, python 3.7,and then:

pip install numpy opencv-python 

pip install torch==1.4.0

pip install torchvision==0.5.0

# run the simple inference script

Take a video of face rotation with a computer camera,and put it into video file

CUDA_VISIBLE_DEVICE=0 python video.py --video_name ./video/your_video_name.mp4

(tips:You can modify video.py file to infer pictures)

/End.

*博客内容为网友个人发布,仅代表博主个人观点,如有侵权请联系工作人员删除。

参与讨论
登录后参与讨论
推荐文章
最近访客