A toolkit for developing and comparing reinforcement learning algorithms.

Datasets, Transforms and Models specific to Computer Vision

A MNIST-like fashion product database. Benchmark 👇

🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐

tensorboard for pytorch (and chainer, mxnet, numpy, ...)

PWC-Net: CNNs for Optical Flow Using Pyramid, Warping, and Cost Volume, CVPR 2018 (Oral)

A clean and readable Pytorch implementation of CycleGAN

[CVPR2024, Highlight] Official code for DragDiffusion

[ICLR2022] official implementation of UniFormer

A high-level toolbox for using complex valued neural networks in PyTorch

Official Pytorch implementation of 6DRepNet: 6D Rotation representation for unconstrained head pose estimation.

CSWin Transformer: A General Vision Transformer Backbone with Cross-Shaped, CVPR 2022

Pytorch implementation of VQGAN (Taming Transformers for High-Resolution Image Synthesis) (https://arxiv.org/pdf/2012.09841.pdf)

Official Code for DiffMorpher: Unleashing the Capability of Diffusion Models for Image Morphing (CVPR 2024)

Code for "Human Pose Regression with Residual Log-likelihood Estimation", ICCV 2021 Oral

Iterative Residual Refinement for Joint Optical Flow and Occlusion Estimation (CVPR 2019)

Toolkits for Multimodal Emotion Recognition

This is the official pytorch implementation for paper: BiDet: An Efficient Binarized Object Detector, which is accepted by CVPR2020.

Combining Faster R-CNN and U-net for efficient medical image segmentation

This is the official Pytorch implementation of "Affine Medical Image Registration with Coarse-to-Fine Vision Transformer" (CVPR 2022), written by Tony C. W. Mok and Albert C. S. Chung.

Official implementation of "Everything at Once - Multi-modal Fusion Transformer for Video Retrieval". CVPR 2022

3D models and code for building your own fingerprint reader

This is the official pytorch implementation for paper: IF-Defense: 3D Adversarial Point Cloud Defense via Implicit Function based Restoration

Distribution-Aware Single-Stage Models for Multi-Person 3D Pose Estimation

Official Github repository for "LiDAR-based Person Re-identification‬". (CVPR 2024)

[Findings of NAACL 2024] Emotion-Anchored Contrastive Learning Framework for Emotion Recognition in Conversation

PyTorch implementation for Contrastive Representation Learning for Gaze Estimation

[MICCAI2023] NICE-Trans: Non-iterative Coarse-to-fine Transformer Networks for Joint Affine and Deformable Image Registration