# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("m3/sscd-copy-detection", dtype="auto")Quick Links
refer: https://github.com/facebookresearch/sscd-copy-detection
# code in src/demo.py
import model
from transformers import pipeline
from transformers.image_utils import load_image
pipe = pipeline(
task='sscd-copy-detection',
model='m3/sscd-copy-detection',
batch_size=10,
device='cpu',
)
vec1 = pipe(load_image("http://images.cocodataset.org/val2017/000000039769.jpg"))
vec2 = pipe(load_image("http://images.cocodataset.org/val2017/000000039769.jpg"))
import torch.nn.functional as F
cos_sim = F.cosine_similarity(vec1, vec2, dim=0)
print('similarity:', cos_sim.item())
- Downloads last month
- 1,690
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-feature-extraction", model="m3/sscd-copy-detection")