【tf-pose-estimation】人体姿态识别tensorflow版OpenPose Ubuntu16.04安装全过程

最新推荐文章于 2021-01-10 19:31:18 发布

原创

最新推荐文章于 2021-01-10 19:31:18 发布 · 4.8k 阅读

33 ·

CC 4.0 BY-SA版权

本文详细记录了在Ubuntu16.04上使用Anaconda3创建虚拟环境，安装CUDA+CUDNN+TensorFlow-GPU，以及解决opencv3安装问题和运行tf-pose-estimation的全流程。通过conda安装避免手动配置CUDA和CUDNN，同时提供了两种opencv3的安装方法，最终成功运行人体姿态识别demo。

3记录一下openpose tf版本的安装全流程，踩了所有的坑，希望你可以一次性安装成功。

环境：Ubuntu16.04+Anaconda3

code：https://github.com/ildoonet/tf-pose-estimation

paper：https://arxiv.org/pdf/1611.08050.pdf

1）创建虚拟环境+安装cuda+cudnn+tensorflow-gpu：

首先创建一个python3.6的虚拟环境，设置conda国内镜像源，然后使用conda来安装tensorflow-gpu，conda安装的好处会自动装上一系列依赖，包括cuda和cudnn（这个前提是安装好了对应显卡的驱动，如何安装驱动有很多教程就不赘述。比起在官网自己下载设置cuda和cudnn，使用conda安装，非常方便。）会自动安装好tensorflow-gpu 1.4.1所对应的cuda和cudnn版本。

conda create -n openposetf python=3.6
source activate openposetf

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls