说明
DeepLabCut是来自EPEL洛桑理工Mackenzie Mathis开发的用于姿态估计开源软件包. 可用少量的数据标注(50-200)得到较为准确的性能. 官网上说目前只支持CUDA10及以下的版本. 另外tensorflow2+要用deeplabcut-core去训练. 这里我安装了两个环境, 一个tensorflow1.3+deeplabcut用来标注, 另一个tensorflow2.0+deeplabcut-core用来训练.
环境
Linux 18.04, CUDA 10.0
安装
- 确保GPU driver 和CUDA安装成功.
- 创建环境, 安装tensorflow
conda create -n DLC python=3.7 tensorflow-gpu=1.13.1
- 在线安装wxPython
pip3 install -U -f https://extras.wxpython.org/wxPython4/extras/linux/gtk3/ubuntu-18.04/ wxPython
- 安装deeplabcut
pip install deeplabcut
- 检测
ipython
import tensorflow
import deeplabcut
没有报错说明安装成功.
教程
image.png
我当时运行label_frame时出现了“wx._core.wxAssertionError: C++ assertion "!(flags & wxALIGN_RIGHT)" failed at /Users/robind/projects/bb2/dist-osx-py37/build/ext/wxWidgets/src/common/sizer.cpp(2098) in DoInsert(): Horizontal alignment flags are ignored in horizontal sizers
”的问题. 对于该问题, 可以尝试安装wxpython==4.0.7.post2
- 先卸载wxPython
pip3 uninstall wxPython
- 找到适合自己python和ubuntu版本的wxpython==4.0.7.post2, 下载
wget https://extras.wxpython.org/wxPython4/extras/linux/gtk3/ubuntu-18.04/wxPython-4.0.7.post2-cp37-cp37m-linux_x86_64.wh
- pip安装
pip3 install wxPython-4.0.7.post2-cp37-cp37m-linux_x86_64.whl
