Repository navigation

vision-transformer-models

Website
Wikipedia

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

Python

33868

4894

11 小时前

yang-ruixin / PyTorch-Image-Models-Multi-Label-Classification

Multi-label classification based on timm.

multi-label multi-label-classification PyTorch pretrained-models pretrained-weights transfer-learning densenet efficientnet hrnet mixnet mnasnet mobilenetv3 regnet resnet timm multi-task-learning vision-transformer-models imagenet-classifier

Jupyter Notebook

118

4 年前

anyantudre / Florence-2-Vision-Language-Model

Florence-2 is a novel vision foundation model with a unified, prompt-based representation for a variety of computer vision and vision-language tasks.

机器视觉深度学习 florence-2 huggingface vision-language vision-language-model vision-transformer vision-transformer-models

Jupyter Notebook

10 个月前

yang-ruixin / pytorch-image-models-with-simclr

Multi-label classification based on timm, and add SimCLR to timm.

simclr semi-supervised-learning self-supervised-learning self-supervised timm PyTorch transfer-learning pretrained-models multi-label-classification multi-label multi-task-learning pretrained-weights efficientnet mixnet mobilenetv3 vision-transformer-models imagenet-classifier

Jupyter Notebook

4 年前

TianyiFranklinWang / MedFM_Submission_Only

Solution for NeurIPS 2023 - MedFM Challenge

人工智能 fine-tuning finetuning foundation-model foundation-models image-classification 机器学习 Python PyTorch vision-transformer vision-transformer-models transfer-learning

Python

2 年前

hadar-hai / vit-vs-cnn-on-elephants

This project focuses on evaluating Convolutional Neural Networks (CNN) and Vision Transformers (ViT) for image classification tasks, specifically distinguishing between Asian elephants and African elephants.

cnn-classification cnn-model 机器视觉深度学习 image-classification PyTorch vision-transformer vision-transformer-models vision-transformers

Jupyter Notebook

1 年前

JaouadT / vision_transformer_pytorch

Code for the base version of the the model vision transformer in pytorch.

PyTorch vision-transformer vision-transformer-models

Python

2 年前