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())

💻 使用示例

基礎用法

# 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())

高級用法

目前文檔未提供高級用法的相關代碼，若有更多需求可參考項目引用的倉庫。

📄 許可證

本項目採用 Apache-2.0 許可證。

🔗 參考鏈接

項目參考了 Facebook Research 的相關倉庫。