Computer Vision

Models

See also: Computer Vision Models

Model	HF Name	Creator	Task	Library	Dataset	Language	Paper	Related to	License
CIDAS/clipseg-rd64-refined model			Image Segmentation	PyTorch Transformers			2112.10003	Vision Clipseg	Apache-2.0
CompVis/stable-diffusion-safety-checker model				PyTorch Transformers			1910.09700 2103.00020	Clip
Facebook/convnext-base-224 model			Image Classification	PyTorch TensorFlow Transformers	Imagenet-1k		2201.03545	Vision AutoTrain Compatible Convnext	Apache-2.0
Facebook/convnext-large-224 model			Image Classification	PyTorch TensorFlow Transformers	Imagenet-1k		2201.03545	Vision AutoTrain Compatible Convnext	Apache-2.0
Google/vit-base-patch16-224 model			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			Image Classification	PyTorch TensorFlow Safetensors Transformers	Imagenet-1k		1512.03385	Vision AutoTrain Compatible Resnet	Apache-2.0
Microsoft/resnet-50 model			Image Classification	PyTorch TensorFlow JAX Transformers	Imagenet-1k		1512.03385	Vision AutoTrain Compatible Resnet	Apache-2.0
Openai/clip-vit-base-patch16 model			Zero-Shot Image Classification	PyTorch JAX Transformers			2103.00020 1908.04913	Clip Vision
Openai/clip-vit-base-patch32 model			Zero-Shot Image Classification	PyTorch TensorFlow JAX Transformers			2103.00020 1908.04913	Clip Vision
Openai/clip-vit-large-patch14 model			Zero-Shot Image Classification	PyTorch TensorFlow JAX Transformers			2103.00020 1908.04913	Clip Vision