Lists (12)
Stars
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.
Unofficial PyTorch implementation of TrackNet
This repository is used for advertising PhD recruitment opportunities. Contributions are welcome!
A minimal implementation of a denoising diffusion model in PyTorch.
"Retinexformer: One-stage Retinex-based Transformer for Low-light Image Enhancement" (ICCV 2023) & (NTIRE 2024 Challenge)
SIFT Flow: Dense Correspondence across Scenes and its Applications. Liu et. al. IEEE transactions on pattern analysis and machine intelligence 33, no. 5 (2010): 978-994.
My implementation of "Patch n’ Pack: NaViT, a Vision Transformer for any Aspect Ratio and Resolution"
[MM'2024] Official implementation of "PEneo: Unifying Line Extraction, Line Grouping, and Entity Linking for End-to-end Document Pair Extraction."
Implementation of Segformer, Attention + MLP neural network for segmentation, in Pytorch
Official PyTorch implementation of SegFormer
This is the official PyTorch implementation of the paper "TransFG: A Transformer Architecture for Fine-grained Recognition" (Ju He, Jie-Neng Chen, Shuai Liu, Adam Kortylewski, Cheng Yang, Yutong Ba…
The official code for “Geometric Representation Learning for Document Image Rectification”, ECCV, 2022.
The repository provides code for running inference with the Meta Segment Anything Model 2 (SAM 2), links for downloading the trained model checkpoints, and example notebooks that show how to use th…
Efficient Multi-modal Models via Stage-wise Visual Context Compression
Official Open Source code for "Masked Autoencoders As Spatiotemporal Learners"
[CVPR 2024] The official pytorch implementation of "A General and Efficient Training for Transformer via Token Expansion".
assistant tools for attention visualization in deep learning
PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
A Graph Attention Spatio-temporal Convolutional Networks for 3D Human Pose Estimation in Video (GAST-Net)
A plug-in replacement for DataLoader to load Imagenet disk-sequentially in PyTorch.
Behavioral cloning: end-to-end learning for self-driving cars.
Third Project of the Udacity Self-Driving Car Nanodegree Program
[ECCV 2022] This is the official implementation of BEVFormer, a camera-only framework for autonomous driving perception, e.g., 3D object detection and semantic map segmentation.
Learning an Isometric Surface Parameterization for Texture Unwrapping (ECCV 2022)
A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.
Official code for DocNLC: A Document Image Enhancement Framework with Normalized and Latent Contrastive Representation for Multiple Degradations (AAAI 2024)