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

📄 ライセンス

このプロジェクトは、Apache-2.0ライセンスの下で公開されています。

🔗 参照

詳細な情報は、こちらのGitHubリポジトリを参照してください。