Models

Pre-trained models

Reference	Name	Purposes	License
torch_arcface_insightface	ArcFace pre-trained by InsightFace	Face recognition	InsightFace
torch_clip_image_encoder	CLIP pre-trained image encoder	Text-to-image retrieval	CLIP
torch_clip_text_encoder	CLIP pre-trained text encoder	Text-to-image retrieval	CLIP
torch_densenet_imagenet	PyTorch DenseNet pre-trained on ImageNet	Image similarity (main subject) Labelling (ImageNet)	TorchVision
torch_densenet_places365	PyTorch DenseNet pre-trained on Places365	Image similarity (context) Labelling (Places365)	Places365
torch_keyframes_resnet_imagenet	Keyframes with ResNet pre-trained on ImageNet	Key-frames (main subject)	TorchVision
torch_keyframes_densenet_imagenet	Keyframes with DenseNet pre-trained on ImageNet	Key-frames (focused on the main subject)	TorchVision
torch_keyframes_densenet_places365	Keyframes with DenseNet pre-trained on Places365	Key-frames (focused on the context)	Places365
torch_magface	Pre-trained MagFace	Face recognition	MagFace
torch_mtcnn	Pre-trained face detection MTCNN from FaceNet	Face detection	FaceNet
torch_resnet_imagenet	PyTorch ResNet pre-trained on ImageNet	Image similarity (main subject) Labelling (ImageNet)	TorchVision
torch_distiluse_base_multilingual_v2	SBERT's pre-trained DistilUse Multilingual Cased V2	Sentence similarity	SBERT
torch_vinvl_detector	VinVL's pre-trained object detector	Object similarity Object detection	Microsoft scene graph benchmark