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有点低,我电脑配置垃圾,总体来说效果还算不错