YubiaoYue

grok-1

Grok open release

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

mmdetection

OpenMMLab Detection Toolbox and Benchmark

CLIP

CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image

deep-learning-for-image-processing

deep learning for image processing including classification and object-detection etc.

peft

🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.

imgaug

Image augmentation for machine learning experiments.

mamba

Mamba SSM architecture

External-Attention-pytorch

🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐

mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

MONAI

AI Toolkit for Healthcare Imaging

pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.

Vim

[ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model

fvcore

Collection of common code that's shared among different research projects in FAIR computer vision team.

torch-cam

Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-CAM)

VMamba

VMamba: Visual State Space Models，code is based on mamba

capsule-networks

A PyTorch implementation of the NIPS 2017 paper "Dynamic Routing Between Capsules".

MedMNIST

[pip install medmnist] 18x Standardized Datasets for 2D and 3D Biomedical Image Classification

Awesome-Mamba-Papers

Awesome Papers related to Mamba.

U-Mamba

U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentation

VM-UNet

(ARXIV24) This is the official code repository for "VM-UNet: Vision Mamba UNet for Medical Image Segmentation".

VisionMamba

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 when performing batch inference to extract features on high-res images

ss-ood

Self-Supervised Learning for OOD Detection (NeurIPS 2019)

Swin-UMamba

Swin-UMamba: Mamba-based UNet with ImageNet-based pretraining

MAE-Lite

Official implement for ICML2023 paper: "A Closer Look at Self-Supervised Lightweight Vision Transformers"

UVM-Net

U-shaped Vision Mamba for Single Image Dehazing

CapsNet

A pytorch implementation of Capsule Network.

USCL

This repository includes the constructed US-4 dataset and the code for USCL (PyTorch version).

OnlineRefer

[ICCV 2023] OnlineRefer: A Simple Online Baseline for Referring Video Object Segmentation

Vision-Mamba-CIFAR10