Image Classification Models

See also: Computer Vision Models and Tasks

Model	Name	User/Org	Task	Library	Dataset	Language	Paper	Related to	License
Facebook/convnext-base-224 model	convnext-base-224	http://facebook	Image Classification	PyTorch TensorFlow Transformers	Imagenet-1k		2201.03545	Vision AutoTrain Compatible Convnext	Apache-2.0
Facebook/convnext-large-224 model	convnext-large-224	http://facebook	Image Classification	PyTorch TensorFlow Transformers	Imagenet-1k		2201.03545	Vision AutoTrain Compatible Convnext	Apache-2.0
Google/vit-base-patch16-224 model	vit-base-patch16-224	http://google	Image Classification	PyTorch TensorFlow JAX Transformers	Imagenet-21k Imagenet-1k		2010.11929 2006.03677	Vision Vit AutoTrain Compatible	Apache-2.0
Microsoft/resnet-18 model	resnet-18	http://microsoft	Image Classification	PyTorch TensorFlow Safetensors Transformers	Imagenet-1k		1512.03385	Vision AutoTrain Compatible Resnet	Apache-2.0
Microsoft/resnet-50 model	resnet-50	http://microsoft	Image Classification	PyTorch TensorFlow JAX Transformers	Imagenet-1k		1512.03385	Vision AutoTrain Compatible Resnet	Apache-2.0