[CVPR 2024 Highlight] Official PyTorch implementation of "MindBridge: A Cross-Subject Brain Decoding Framework"

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

OpenMMLab Self-Supervised Learning Toolbox and Benchmark

Implementation of Zero-Shot Image-to-Text Generation for Visual-Semantic Arithmetic

Official PyTorch implementation of "CleanCLIP: Mitigating Data Poisoning Attacks in Multimodal Contrastive Learning" @ ICCV 2023

The example of correspondence between fine classes and superclass (coarse class) in ImageNet.

A library for experimenting with, training and evaluating neural networks, with a focus on adversarial robustness.

Versatile Diffusion: Text, Images and Variations All in One Diffusion Model, arXiv 2022 / ICCV 2023

Official Implementation for NeurIPS'23 paper Contrast, Attend and Diffuse to Decode High-Resolution Images from Brain Activities

Official repo for MindArt

PyTorch Implementation of the ICCV 2023 paper: Generalized Differentiable RANSAC ($\nabla$-RANSAC).

PyTorch implementation of a Variational Autoencoder with Gumbel-Softmax Distribution

categorical variational autoencoder using the Gumbel-Softmax estimator

Code for Optimal Transport for structured data with application on graphs

Geometric loss functions between point clouds, images and volumes

CO-Optimal Transport

code for paper "Improving Sequence-to-Sequence Learning via Optimal Transport"

[WACV 2024] DREAM: Visual Decoding from Reversing Human Visual System

GIM: Learning Generalizable Image Matcher From Internet Videos (ICLR 2024 Spotlight)

fMRI-to-image reconstruction on the NSD dataset.

Official PyTorch implementation of GroupViT: Semantic Segmentation Emerges from Text Supervision, CVPR 2022.

Code for ICML 2020 "Graph Optimal Transport for Cross-Domain Alignment"

Code for Sliced Gromov-Wasserstein

A Python Graph Matching Toolkit.

Compute FID scores with PyTorch.

🤗 Evaluate: A library for easily evaluating machine learning models and datasets.

Learning to Transform for Generalizable Instance-wise Invariance