Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

[ICCV 2021 Oral] PoinTr: Diverse Point Cloud Completion with Geometry-Aware Transformers

[ICLR'24 Spotlight] Uni3D: 3D Visual Representation from BAAI

Official PyTorch implementation of Fully Attentional Networks

Official implementation for the paper "Deep ViT Features as Dense Visual Descriptors".

EsViT: Efficient self-supervised Vision Transformers

[NIVT Workshop @ ICCV 2023] SeMask: Semantically Masked Transformers for Semantic Segmentation

SimpleClick: Interactive Image Segmentation with Simple Vision Transformers (ICCV 2023)

Implementation of MetNet-3, SOTA neural weather model out of Google Deepmind, in Pytorch

Implementation of CrossViT: Cross-Attention Multi-Scale Vision Transformer for Image Classification

[ECCV 2022] The official repo for the paper "Poseur: Direct Human Pose Regression with Transformers".

A Monocular depth-estimation for in-the-wild AutoFocus application.

Official repository for "Self-Supervised Video Transformer" (CVPR'22)

Few-Shot Diffusion Models

Determine whether a given video sequence has been manipulated or synthetically generated

[ECCV 2024] Pytorch code for our ECCV'24 paper NeRF-MAE: Masked AutoEncoders for Self-Supervised 3D Representation Learning for Neural Radiance Fields

[NeurIPS'21] "Chasing Sparsity in Vision Transformers: An End-to-End Exploration" by Tianlong Chen, Yu Cheng, Zhe Gan, Lu Yuan, Lei Zhang, Zhangyang Wang

Rethinking Interactive Image Segmentation with Low Latency, High Quality, and Diverse Prompts (CVPR 2024)

Official implement of Evo-ViT: Slow-Fast Token Evolution for Dynamic Vision Transformer

[ECCV 2022] "PPT: token-Pruned Pose Transformer for monocular and multi-view human pose estimation"