Downloads · 30 days
18
9% of all-time downloads
aeth0r/cat2vec-v2
cat2vec-v2 is a image feature extraction model from aeth0r. Use it for the image feature extraction task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as gpl-3.0.
The cat2vec model is a search model for cats.
Downloads · 30 days
18
9% of all-time downloads
All-time downloads
197
Public
Parameters
22.1M
88.2 MB on disk
Likes
1
Public
Click a slice to open those files.
.safetensors88.2 MB · 100%
From the Hugging Face model README
The cat2vec model is a search model for cats.
It was trained using the Labeled Cats In The Wild dataset and a triplet loss.
from transformers import AutoImageProcessor, ResNetModel
import torch
from datasets import load_dataset
dataset = load_dataset("huggingface/cats-image")
image = dataset["test"]["image"][:2]
processor = AutoImageProcessor.from_pretrained("facebook/dinov2-small")
model = ResNetModel.from_pretrained("aeth0r/cat2vec-v2")
inputs = processor(image, return_tensors="pt")
with torch.no_grad():
features = model(**inputs)
print(features)