This repo is about implementing different approaches of pose estimation and also is a sub-task of the smart hospital bed project :smile:

Last update: Oct 17, 2022

Overview

Pose-Estimation

This repo is a sub-task of the smart hospital bed project which is about implementing the task of pose estimation 😄 Many thanks to the main authors of hrnet-human-pose-estimation project - the project I'd mainly based on.

Setup

Data preparation

I used coco dataset(2017) which can be downloaded here, and the format is just like this:

${POSE_ROOT}
|-- data
`-- |-- coco
    `-- |-- annotations
        |   |-- person_keypoints_train2017.json
        |   `-- person_keypoints_val2017.json
        |-- person_detection_results
        |   |-- COCO_val2017_detections_AP_H_56_person.json
        |   |-- COCO_test-dev2017_detections_AP_H_609_person.json
        `-- images
            |-- train2017
            |   |-- 000000000009.jpg
            |   |-- 000000000025.jpg
            |   |-- 000000000030.jpg
            |   |-- ... 
            `-- val2017
                |-- 000000000139.jpg
                |-- 000000000285.jpg
                |-- 000000000632.jpg
                |-- ...

Dependencies

Here are required packages:

torch
torchvision
numpy
json_tricks
yacs>=0.1.5
Cython

You also need to install cocoapi:

git clone https://github.com/cocodataset/cocoapi.git
cd cocoapi/PythonAPI
# Install into global site-packages
make install
# Alternatively, if you do not have permissions or prefer
# not to install the COCO API into global site-packages
python3 setup.py install --user
cd ..

Usage

pip install -r requirements.txt
cd src
python3 main.py # update later

Citation

@inproceedings{sun2019deep,
  title={Deep High-Resolution Representation Learning for Human Pose Estimation},
  author={Sun, Ke and Xiao, Bin and Liu, Dong and Wang, Jingdong},
  booktitle={CVPR},
  year={2019}
}

@inproceedings{xiao2018simple,
    author={Xiao, Bin and Wu, Haiping and Wei, Yichen},
    title={Simple Baselines for Human Pose Estimation and Tracking},
    booktitle = {European Conference on Computer Vision (ECCV)},
    year = {2018}
}

This repo is about implementing different approaches of pose estimation and also is a sub-task of the smart hospital bed project :smile:

Related tags

Overview

Pose-Estimation

Setup

Data preparation

Dependencies

Usage

Citation

Owner

Max

PyTorch implementations of neural network models for keyword spotting

OOD Dataset Curator and Benchmark for AI-aided Drug Discovery

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

Fast and Easy Infinite Neural Networks in Python

Distributed Evolutionary Algorithms in Python

CLDF dataset derived from Robbeets et al.'s "Triangulation Supports Agricultural Spread" from 2021

Shared Attention for Multi-label Zero-shot Learning

Torch implementation of SegNet and deconvolutional network

Notes taking website build with Docker + Django + React.

Generative Flow Networks

A Flexible Generative Framework for Graph-based Semi-supervised Learning (NeurIPS 2019)

Clockwork Convnets for Video Semantic Segmentation

This is an official implementation for "PlaneRecNet".

This repository contains the code for our fast polygonal building extraction from overhead images pipeline.

A platform for intelligent agent learning based on a 3D open-world FPS game developed by Inspir.AI.

RNN Predict Street Commercial Vitality

Tidy interface to polars

Vision-Language Pre-training for Image Captioning and Question Answering

Implementation of the bachelor's thesis "Real-time stock predictions with deep learning and news scraping".

The code for our paper Semi-Supervised Learning with Multi-Head Co-Training