PyTorch implementation of the Pose Residual Network (PRN)

Last update: Nov 28, 2022

Overview

Pose Residual Network

This repository contains a PyTorch implementation of the Pose Residual Network (PRN) presented in our ECCV 2018 paper:

Muhammed Kocabas, Salih Karagoz, Emre Akbas. MultiPoseNet: Fast Multi-Person Pose Estimation using Pose Residual Network. In ECCV, 2018. arxiv

PRN is described in Section 3.2 of the paper.

Getting Started

We have tested our method on Coco Dataset

Prerequisites

python
pytorch
numpy
tqdm
pycocotools
progress
scikit-image

Installing

Clone this repository git clone https://github.com/salihkaragoz/pose-residual-network-pytorch.git
Install Pytorch
pip install -r src/requirements.txt
To download COCO dataset train2017 and val2017 annotations run: bash data/coco.sh. (data size: ~240Mb)

Training

python train.py

For more options look at opt.py

Testing

Download pre-train model
python test.py --test_cp=PathToPreTrainModel/PRN.pth.tar

Results

Results on COCO val2017 Ground Truth data.

 Average Precision  (AP) @[ IoU=0.50:0.95 | area=   all | maxDets= 20 ] = 0.892
 Average Precision  (AP) @[ IoU=0.50      | area=   all | maxDets= 20 ] = 0.978
 Average Precision  (AP) @[ IoU=0.75      | area=   all | maxDets= 20 ] = 0.921
 Average Precision  (AP) @[ IoU=0.50:0.95 | area=medium | maxDets= 20 ] = 0.883
 Average Precision  (AP) @[ IoU=0.50:0.95 | area= large | maxDets= 20 ] = 0.912
 Average Recall     (AR) @[ IoU=0.50:0.95 | area=   all | maxDets= 20 ] = 0.917
 Average Recall     (AR) @[ IoU=0.50      | area=   all | maxDets= 20 ] = 0.982
 Average Recall     (AR) @[ IoU=0.75      | area=   all | maxDets= 20 ] = 0.937
 Average Recall     (AR) @[ IoU=0.50:0.95 | area=medium | maxDets= 20 ] = 0.902
 Average Recall     (AR) @[ IoU=0.50:0.95 | area= large | maxDets= 20 ] = 0.944

License

Citation

If you find this code useful for your research, please consider citing our paper:

@Inproceedings{kocabas18prn,
  Title          = {Multi{P}ose{N}et: Fast Multi-Person Pose Estimation using Pose Residual Network},
  Author         = {Kocabas, Muhammed and Karagoz, Salih and Akbas, Emre},
  Booktitle      = {European Conference on Computer Vision (ECCV)},
  Year           = {2018}
}

PyTorch implementation of the Pose Residual Network (PRN)

Related tags

Overview

Pose Residual Network

Getting Started

Prerequisites

Installing

Training

Testing

Results

License

Citation

Owner

Salih Karagoz

A clean and robust Pytorch implementation of PPO on continuous action space.

Unofficial implementation of Proxy Anchor Loss for Deep Metric Learning

PyTorch Implementation of "Non-Autoregressive Neural Machine Translation"

Source code and notebooks to reproduce experiments and benchmarks on Bias Faces in the Wild (BFW).

CVPR2021 Workshop - HDRUNet: Single Image HDR Reconstruction with Denoising and Dequantization.

Object detection (YOLO) with pytorch, OpenCV and python

Official Implementation of Neural Splines

ROS Basics and TurtleSim

Source code for CVPR2022 paper "Abandoning the Bayer-Filter to See in the Dark"

Informal Persian Universal Dependency Treebank

[ICCV 2021 Oral] NerfingMVS: Guided Optimization of Neural Radiance Fields for Indoor Multi-view Stereo

Pytorch ImageNet1k Loader with Bounding Boxes.

Reinforcement learning algorithms in RLlib

Real-time VIBE: Frame by Frame Inference of VIBE (Video Inference for Human Body Pose and Shape Estimation)

SlotRefine: A Fast Non-Autoregressive Model forJoint Intent Detection and Slot Filling

Neural style in TensorFlow! 🎨

Pytorch-diffusion - A basic PyTorch implementation of 'Denoising Diffusion Probabilistic Models'

A deep-learning pipeline for segmentation of ambiguous microscopic images.

PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR) and Generative Adversarial Imitation Learning (GAIL).

Adaptable tools to make reinforcement learning and evolutionary computation algorithms.