Official implementation of the CVPR 2024 paper ViT-CoMer: Vision Transformer with Convolutional Multi-scale Feature Interaction for Dense Predictions.

Multi-lingual large voice generation model, providing inference, training and deployment full-stack ability.

The official PyTorch implementation of L2CS-Net for gaze estimation and tracking

Official implementation of ETH-XGaze dataset baseline

YOLOv10: Real-Time End-to-End Object Detection [NeurIPS 2024]

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

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

Visual tracking library based on PyTorch.

Code for "OnePose++: Keypoint-Free One-Shot Object Pose Estimation without CAD Models" NeurIPS 2022

[CVPR 2019] Learning Parallax Attention for Stereo Image Super-Resolution

[TPAMI'23] Unifying Flow, Stereo and Depth Estimation

YOLOv5的轻量化改进(蜂巢检测项目)

⚡ A newly designed ultra lightweight anchor free target detection algorithm， weight only 250K parameters， reduces the time consumption by 10% compared with yolo-fastest, and the post-processing is …

The implementation of the paper 'ZebraPose: Coarse to Fine Surface Encoding for 6DoF Object Pose Estimation' (CVPR2022)

PyTorch Implementation of DeepVO

Spatial Temporal Graph Convolutional Networks (ST-GCN) for Skeleton-Based Action Recognition in PyTorch

[ECCV2020 Oral] Learning Lane Graph Representations for Motion Forecasting

A microscopic traffic simulation in Python

Pytorch implementation of CVPR2020 paper “VectorNet: Encoding HD Maps and Agent Dynamics from Vectorized Representation”

Custom implementation of TNT-VectorNet and HOME architectures

[CVPR 2023] Query-Centric Trajectory Prediction

Vectornet for trajectory prediction, implemented in PyTorch/Torch_geometric

Official GitHub repository for Argoverse dataset

Graph Attention Networks (https://arxiv.org/abs/1710.10903)

Pytorch implementation of the Graph Attention Network model by Veličković et. al (2017, https://arxiv.org/abs/1710.10903)

The pytorch implementation of DG-Font: Deformable Generative Networks for Unsupervised Font Generation