cuda和cudnn的教程网上一堆,自己装好

本教程的cuda版本为

python版本3.5.2 tf版本1.7.1

这里上一个作者的环境搭建视频

https://www.youtube.com/watch?v=nUjGLjOmF7o

cmd输入python,再输入import tensorflow as tf,没报错的话,恭喜你,可以开始本教程了。


先上tensorflow版本openpose的地址

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

下载解压到任意路径下,建议路径不要太长,我的直接放在D盘

cmd定位到你的tf-pose-estimation-master文件夹

运行

pip install -r requirements.txt
 

requirements.txt文件内容如下

运行这个txt需要安装git,之前安装过的可以跳过这一节


https://git-scm.com/downloads/

点击显示屏里的连接下载

直接打开exe,一路next

安装成功后会出现桌面快捷方式,或者鼠标右击桌面

打开git bash,输入git,至此git安装完成,回到前面说的pip install -r requirements.txt

等待安装完成后运行py。好的,一堆报错大概有两个报错

ImportError: No module named '_pafprocess'

NameError: name 'ModuleNotFoundError' is not defined

这也是最坑的地方。找了很多issues才找到答案

我们看看官方是怎么说的

没错,我们还要用swig编译一次

ok,先来下swig,同样之前下过的可以跳过这一节


我的swig版本是swigwin-3.0.12,下载地址是http://www.swig.org/download.html

解压zip,比如:D:\backupsoftware

添加环境变量到path, 比如: D:\backupsoftware\swigwin-3.0.12

在命令行执行: swig --help,不报错说明安装成功了。


下面开始swig代码,进入tf-pose-estimation-master文件夹下,cmd定位到D:\tf-pose-estimation\tf_pose\pafprocess

在Tensorflow环境下,运行

swig -python -c++ pafprocess.i && python setup.py build_ext --inplace

如果不报错,你的文件夹内容应该是这样的

如果报错了,恭喜你,又中奖了,而且我猜你的错误应该是这个

Unable to find vcvarsall.bat

首先找到你的vcvarsall.bat所在路径。也就是这个文件

此文件一般位于vs安装路径下,我的路径:\Program Files (x86)\Microsoft Visual Studio\2017\Community\VC\Auxiliary\Build

接着说一个我们需要让python找到这个文件,这里只介绍我成功的方法,如果不行,百度百度百度

首先找到_msvccompiler.py文件,我的路径是C:\Users\work-25\Anaconda3\envs\tensorflow-gpu\Lib\distutils

你们在自己的python文件下面找找。打开这个文件,找到

def _find_vcvarsall(plat_spec):
 

没修改前,这个方法是这样的

def _find_vcvarsall(plat_spec):
    try:
        key = winreg.OpenKeyEx(
            winreg.HKEY_LOCAL_MACHINE,
            r"Software\Microsoft\VisualStudio\SxS\VC7",
            access=winreg.KEY_READ | winreg.KEY_WOW64_32KEY
        )
    except OSError:
        log.debug("Visual C++ is not registered")
        return None, None
 
    with key:
        best_version = 0
        best_dir = None
        for i in count():
            try:
                v, vc_dir, vt = winreg.EnumValue(key, i)
            except OSError:
                break
            if v and vt == winreg.REG_SZ and os.path.isdir(vc_dir):
                try:
                    version = int(float(v))
                except (ValueError, TypeError):
                    continue
                if version >= 14 and version > best_version:
                    best_version, best_dir = version, vc_dir
        if not best_version:
            log.debug("No suitable Visual C++ version found")
            return None, None
 
        vcvarsall = os.path.join(best_dir, "vcvarsall.bat")
        if not os.path.isfile(vcvarsall):
            log.debug("%s cannot be found", vcvarsall)
            return None, None
 
        vcruntime = None
        vcruntime_spec = _VCVARS_PLAT_TO_VCRUNTIME_REDIST.get(plat_spec)
        if vcruntime_spec:
            vcruntime = os.path.join(best_dir,
                vcruntime_spec.format(best_version))
            if not os.path.isfile(vcruntime):
                log.debug("%s cannot be found", vcruntime)
                vcruntime = None
 
        return vcvarsall, vcruntime

修改后是这样的

def _find_vcvarsall(plat_spec):
        best_dir = r'C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\VC\Auxiliary\Build'
        best_version = 17
        vcruntime = None
        vcruntime_spec = _VCVARS_PLAT_TO_VCRUNTIME_REDIST.get(plat_spec)
        if vcruntime_spec:
            vcruntime = os.path.join(best_dir,
                vcruntime_spec.format(best_version))
            if not os.path.isfile(vcruntime):
                log.debug("%s cannot be found", vcruntime)
                vcruntime = None
        print(vcruntime)
        return r'C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\VC\Auxiliary\Build\vcvarsall.bat', vcruntime

再次回到pafprocess文件夹运行

swig -python -c++ pafprocess.i && python setup.py build_ext --inplace
 

不报错就成功了,到这里你就可以成功使用tf-openpose了

定位到D:\tf-pose-estimation,运行

python run.py --model=mobilenet_thin --resize=432x368 --image=./images/p1.jpg
 

来张骚猪和吴秀波(孙狗)的斗舞

fps有点低,我电脑配置垃圾,总体来说效果还算不错

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐