search

openpifpaf / openpifpaf

Official implementation of "OpenPifPaf: Composite Fields for Semantic Keypoint Detection and Spatio-Temporal Association" in PyTorch.
https://openpifpaf.github.io
Other
1.17k stars 247 forks source link

Support torch versions 2+ #630

Open dsantiago opened 5 months ago

dsantiago commented 5 months ago

It's hard to coexist this project with other because the environment must be almost made only for this project as torch versions is still 1.x.x

Any plans on update it

yangperasd commented 1 month ago

you can compile and install by yourself instead wheel. try something below:

git clone https://github.com/openpifpaf/openpifpaf.git cd openpifpaf // remove torch==1.13.1 torchvision==0.14.1 numpy>=1.16,<2 in install_requires vim setup.py python setup.py install

i tested on pytorch=2.4.0+cu118