DUSt3R: Geometric 3D Vision Made Easy

PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

Official implementation of the NeurIPS 2023 paper "Self-supervised Object-Centric Learning for Videos"

PyTorch Tutorial on google colaboratory.

[CVPR 2024] Panda-70M: Captioning 70M Videos with Multiple Cross-Modality Teachers

PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO

Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch

iBOT 🤖: Image BERT Pre-Training with Online Tokenizer (ICLR 2022)

PyTorch code and models for the DINOv2 self-supervised learning method.

Development kit for TrackingNet

Visual localization made easy with hloc

[EMNLP 2023 (Findings)] This repository contains data processing, evaluation, and fine-tuning code for NEWTON: Are Large Language Models Capable of Physical Reasoning?

[ICCV 2023] PoseDiffusion: Solving Pose Estimation via Diffusion-aided Bundle Adjustment

Vector (and Scalar) Quantization, in Pytorch

OmniGibson: a platform for accelerating Embodied AI research built upon NVIDIA's Omniverse engine. Join our Discord for support: https://discord.gg/bccR5vGFEx

Calculate Root-mean-square deviation (RMSD) of two molecules, using rotation, in xyz or pdb format

Python bindings for COLMAP

Code for RelPose (ECCV 2022)

Video datasets

Tracking Any Point (TAP)

Metric depth estimation from a single image

train test labels for sunrgbd

ADE20K Dataset

The official implementation of Autoregressive Image Generation using Residual Quantization (CVPR '22)