Computer Vision

Models

See also: Computer Vision Models

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