Computer Vision

Models

See also: Computer Vision Models

Model	Name	User/Org	Task	Library	Dataset	Language	Paper	Related to	License
CIDAS/clipseg-rd64-refined model	clipseg-rd64-refined	http://CIDAS	Image Segmentation	PyTorch Transformers			2112.10003	Vision Clipseg	Apache-2.0
CompVis/stable-diffusion-safety-checker model	stable-diffusion-safety-checker	http://CompVis		PyTorch Transformers			1910.09700 2103.00020	Clip
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
Openai/clip-vit-base-patch16 model	clip-vit-base-patch16	http://openai	Zero-Shot Image Classification	PyTorch JAX Transformers			2103.00020 1908.04913	Clip Vision
Openai/clip-vit-base-patch32 model	clip-vit-base-patch32	http://openai	Zero-Shot Image Classification	PyTorch TensorFlow JAX Transformers			2103.00020 1908.04913	Clip Vision
Openai/clip-vit-large-patch14 model	clip-vit-large-patch14	http://openai	Zero-Shot Image Classification	PyTorch TensorFlow JAX Transformers			2103.00020 1908.04913	Clip Vision