Lists (1)
Stars
Denoising Diffusion Probabilistic Models
DreamDA: Generative Data Augmentation with Diffusion Models (Official Implementation)
LPIPS metric. pip install lpips
Official PyTorch implementation of SegFormer
[CVPR 2022--Oral] Restormer: Efficient Transformer for High-Resolution Image Restoration. SOTA for motion deblurring, image deraining, denoising (Gaussian/real data), and defocus deblurring.
A curated list of resources for Image and Video Deblurring
[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
[ECCV 2024] Official implementation of the paper "Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection"
A modified Unet for single image super resolution
This is a warehouse for MobileNetV4-Pytorch-model, can be used to train your image-datasets for vision tasks.
An unofficial implementation of MobileNetV4 in Pytorch
(ICCV'23) Learning to Upsample by Learning to Sample
RepViT: Revisiting Mobile CNN From ViT Perspective [CVPR 2024] and RepViT-SAM: Towards Real-Time Segmenting Anything
Pytorch implementation of "LightM-UNet: Mamba Assists in Lightweight UNet for Medical Image Segmentation"
RUNet: A Robust UNet Architecture for Image Super-Resolution
[Patterns (Cell subsidiary)] The official code for "UltraLight VM-UNet: Parallel Vision Mamba Significantly Reduces Parameters for Skin Lesion Segmentation".
Implementation of Vision Mamba from the paper: "Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model" It's 2.8x faster than DeiT and saves 86.8% GPU memory wh…
(ARXIV24) This is the official code repository for "VM-UNet: Vision Mamba UNet for Medical Image Segmentation".
[ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model
Official implementation of the paper "DeblurDiNAT: A Generalizable Transformer for Perceptual Image Deblurring".
[ECCV2022] Efficient Long-Range Attention Network for Image Super-resolution