代码实践(干货)| 基于Pytorch的人脸姿态检测(附源码)
转载
计算机视觉研究院
”
计算机视觉研究院专栏
作者:Edison_G
—— 01 ——
—— 02 ——
部分代码如下:
部分代码如下:
—— 03 ——
模型的调用和显示
(1)根据点坐标绘制标注图形:
部分代码如下:
部分代码如下:
效果如下图所示:
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.
计算机视觉研究院主要涉及深度学习领域,主要致力于人脸检测、人脸识别,多目标检测、目标跟踪、图像分割等研究方向。研究院接下来会不断分享最新的论文算法新框架,我们这次改革不同点就是,我们要着重”研究“。之后我们会针对相应领域分享实践过程,让大家真正体会摆脱理论的真实场景,培养爱动手编程爱动脑思考的习惯!
后台回复“人脸姿态”
获取实践源码下载地址
计算机视觉研究院
长按扫描二维码关注我们