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

KwaiRec: A Fully-observed Dataset for Recommender Systems (Density: Almost 100%)

Code for "Primitive Representation Learning for Scene Text Recognition" (CVPR 2021)

Per-Pixel Classification is Not All You Need for Semantic Segmentation

The official PyTorch code for NeurIPS 2021 ML4AD Paper, "Does Thermal data make the detection systems more reliable?"

Sub-Cluster AdaCos: Learning Representations for Anomalous Sound Detection.

Omniverse sample scripts - A guide for developing with Python scripts on NVIDIA Ominverse

Official PyTorch implementation for Generic Attention-model Explainability for Interpreting Bi-Modal and Encoder-Decoder Transformers, a novel method to visualize any Transformer-based network. Including examples for DETR, VQA.

Multi-View Radar Semantic Segmentation

Python scripts using the Mediapipe models for Halloween.

Multitask Learning Strengthens Adversarial Robustness

Detectron2-FC a fast construction platform of neural network algorithm based on detectron2

fcn by tensorflow

Uncertainty-aware Semantic Segmentation of LiDAR Point Clouds for Autonomous Driving

Repository for XLM-T, a framework for evaluating multilingual language models on Twitter data

Neural Scene Flow Fields for Space-Time View Synthesis of Dynamic Scenes

Implementation of Kalman Filter in Python

Python package for dynamic system estimation of time series

Omnidirectional Scene Text Detection with Sequential-free Box Discretization (IJCAI 2019). Including competition model, online demo, etc.

Additional environments compatible with OpenAI gym

Shared Attention for Multi-label Zero-shot Learning