Mask R-CNN and Keypoint R-CNN api wrapper in PyTorch.
- Install PyTorch
>= 1.2
with torchvision>= 0.4
pip install git+https://github.com/prasunroy/rcnnpose-pytorch.git
git clone https://github.com/prasunroy/rcnnpose-pytorch.git
cd rcnnpose-pytorch
python setup.py install
cd examples
python image_demo.py
python video_demo.py
Made with ❤️ and 🍕