Highlights
- Pro
Stars
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
Time series forecasting with PyTorch
Code for "LoFTR: Detector-Free Local Feature Matching with Transformers", CVPR 2021, T-PAMI 2022
[CVPR 2024 Highlight] FoundationPose: Unified 6D Pose Estimation and Tracking of Novel Objects
The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.
PyTorch code and models for the DINOv2 self-supervised learning method.
Welcome to the project repository for POPE (Promptable Pose Estimation), a state-of-the-art technique for 6-DoF pose estimation of any object in any scene using a single reference.
An Open Platform for Population Synthesis
Pytorch Implementation for Stepwise Goal-Driven Networks for Trajectory Prediction (RA-L/ICRA2022)
The code for Bi-directional Trajectory Prediction (BiTraP).
Annotation data for JAAD (Joint Attention in Autonomous Driving) Dataset
Annotations for Pedestrian Intention Estimation (PIE) dataset