search

MVIG-SJTU / AlphaPose

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System
http://mvig.org/research/alphapose.html
Other
7.83k stars 1.96k forks source link

Succeeded in Building AlphaPose [py 3.10, pt 1.12, cuda 11.3] [ubuntu 22.04] #1206

Open jamesnulliu opened 3 months ago

jamesnulliu commented 3 months ago 
# Create virtual env
conda create -n alphapose python=3.10 -y && conda activate alphapose
# Install pytorch 1.12.0 with cuda 11.3
pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu113

Since there is a bug in HapleCOCOAPI, it has to be installed with following commands:

# HalpeCOCOAPI 
git clone -b patch-import-path --single-branch git@github.com:bmmtstb/HalpeCOCOAPI.git HalpeCOCOAPI
# Install
cd HalpeCOCOAPI/PythonAPI && python setup.py build develop

Clone the repo and install AlphaPose:

# Clone the repo
git clone git@github.com:MVIG-SJTU/AlphaPose.git AlphaPose && cd AlphaPose
# Install
pip install cython matplotlib cython_bbox numpy scipy easydict opencv-python pycocotools pyyaml tensorboardx terminaltables tqdm visdom
CC=gcc-9 CXX=g++-9 python setup.py build develop
ChataingT commented 3 days ago

Thanks for the help.
Here is my working docker published : tchataing/alphapose
and its dockerfile https://github.com/ChataingT/alphapose_docker.git