OpenMMLab Pose Estimation Toolbox and Benchmark.

(CVPR 2023) Pytorch implementation of “T2M-GPT: Generating Human Motion from Textual Descriptions with Discrete Representations”

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

Vector Quantization, in Pytorch

An easy to use, user-friendly and efficient code for extracting OpenAI CLIP (Global/Grid) features from image and text respectively.

A Collection of Variational Autoencoders (VAE) in PyTorch.

[CVPR 2022] MHFormer: Multi-Hypothesis Transformer for 3D Human Pose Estimation

A repository to host extended examples and tutorials

Datasets, Transforms and Models specific to Computer Vision

Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorch

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

95.47% on CIFAR10 with PyTorch

This repo is copied from https://github.com/leoxiaobin/deep-high-resolution-net.pytorch

Efficient 3D human pose estimation in video using 2D keypoint trajectories

Gradually-Warmup Learning Rate Scheduler for PyTorch

Anatomy-aware 3D Human Pose Estimation in Videos

Flamegraph generator for cProfile stats

The Pytorch implementation for "Semantic Graph Convolutional Networks for 3D Human Pose Regression" (CVPR 2019).

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

The OCR approach is rephrased as Segmentation Transformer: https://arxiv.org/abs/1909.11065. This is an official implementation of semantic segmentation for HRNet. https://arxiv.org/abs/1908.07919

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

Project and dataset webpage: