RCNNPose PyTorch

Mask R-CNN and Keypoint R-CNN api wrapper in PyTorch.

Installation

Prerequisites

• Install PyTorch >= 1.2 with torchvision >= 0.4

Option 1: Install using pip

pip install git+https://github.com/prasunroy/rcnnpose-pytorch.git

Option 2: Install from source

git clone https://github.com/prasunroy/rcnnpose-pytorch.git
cd rcnnpose-pytorch
python setup.py install

Running examples

cd examples
python image_demo.py
python video_demo.py

License

• RCNNPose PyTorch License

Made with ❤️ and 🍕