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

Sample Prior Guided Robust Model Learning to Suppress Noisy Labels

Official Pytorch implementation for video neural representation (NeRV)

pix2pix in tensorflow.js

SNIPS: Solving Noisy Inverse Problems Stochastically

A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks

Learning based AI for playing multi-round Koi-Koi hanafuda card games. Have fun.

Code for paper Novel View Synthesis via Depth-guided Skip Connections

Official implementation of Deep Burst Super-Resolution

A PyTorch implementation of the paper Mixup: Beyond Empirical Risk Minimization in PyTorch

Price-Prediction-For-a-Dream-Home - A machine learning based linear regression trained model for house price prediction.

Poplar implementation of "Bundle Adjustment on a Graph Processor" (CVPR 2020)

Interactive web apps created using geemap and streamlit

Aerial Single-View Depth Completion with Image-Guided Uncertainty Estimation (RA-L/ICRA 2020)

Custom studies about block sparse attention.

Context-Aware Image Matting for Simultaneous Foreground and Alpha Estimation

PyTorch implementation of HDN(Homography Decomposition Networks) for planar object tracking

An unofficial PyTorch implementation of a federated learning algorithm, FedAvg.

Sample code from the Neural Networks from Scratch book.

CoReD: Generalizing Fake Media Detection with Continual Representation using Distillation (ACMMM'21 Oral Paper)

Fader Networks: Manipulating Images by Sliding Attributes - NIPS 2017