许可协议：Apache-2.0
标签：

• 图像特征提取

参考：https://github.com/facebookresearch/sscd-copy-detection

# 源代码位于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('相似度:', cos_sim.item())