vision-transformer-models

There are 2 repositories under vision-transformer-models topic.

huggingface / 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), MobileNet-V3/V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
augmix convnext distributed-training dual-path-networks efficientnet image-classification imagenet maxvit mixnet mobile-deep-learning mobilenet-v2 mobilenetv3 nfnets normalization-free-training pretrained-models pretrained-weights pytorch randaugment resnet vision-transformer-models
Language:Python 30061
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 gradient-centralization densenet efficientnet hrnet mixnet mnasnet mobilenetv3 regnet resnet multi-task timm multi-task-learning vision-transformer-models imagenet-classifier
Language:Jupyter Notebook 102
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 semi-supervised self-supervised timm pytorch transfer-learning pretrained-models multi-label-classification multi-label multi-task multi-task-learning pretrained-weights efficientnet gradient-centralization mixnet mobilenetv3 vision-transformer-models imagenet-classifier
Language:Jupyter Notebook 35
TianyiFranklinWang / MedFM_Submission_Only
Solution for NeurIPS 2023 - MedFM Challenge
artificial-intelligence fine-tuning finetuning foundation-model foundation-models image-classification image-classifier machine-learning machine-learning-algorithms python pytorch vision-transformer vision-transformer-image-classification vision-transformer-models medfm medfm-competition medical-image-classification neurips-2023 transfer-learning
Language:Python 1
JaouadT / vision_transformer_pytorch
Code for the base version of the the model vision transformer in pytorch.
pytorch vision-transformer vision-transformer-models
Language:Python 0
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 computer-vision deep-learning image-classification machine-learnign pytorch vision-transformer vision-transformer-image-classification vision-transformer-models vision-transformers
Language:Jupyter Notebook