计算机视觉研究院


代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_ide

计算机视觉研究院专栏

作者:Edison_G

代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_计算机视觉_02



代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_python_03

代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_计算机视觉_04

代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_python_05


代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_ide_06


—— 01 ——

代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_计算机视觉_07


—— 02 ——

代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_计算机视觉_08

部分代码如下:

代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_计算机视觉_09

代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_计算机视觉_10

代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_ide_11

部分代码如下:

代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_ide_12


—— 03 ——

模型的调用和显示

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

部分代码如下:

代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_python_13

代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_计算机视觉_14

部分代码如下:

代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_计算机视觉_15

效果如下图所示:

代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_ide_16


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.



计算机视觉研究院主要涉及深度学习领域,主要致力于人脸检测、人脸识别,多目标检测、目标跟踪、图像分割等研究方向。研究院接下来会不断分享最新的论文算法新框架,我们这次改革不同点就是,我们要着重”研究“。之后我们会针对相应领域分享实践过程,让大家真正体会摆脱理论的真实场景,培养爱动手编程爱动脑思考的习惯!

后台回复“人脸姿态”

代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_计算机视觉_17

获取实践源码下载地址

代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)_计算机视觉_18

计算机视觉研究院

长按扫描二维码关注我们