Views
No views yet
pip install -U sentence-transformers fugashi SentencePiece 1from sentence_transformers import SentenceTransformer
2model = SentenceTransformer("oshizo/japanese-clip-qwen2_vl-exp-0126", trust_remote_code=True)
3
4import io
5import requests
6from PIL import Image
7
8sentences = [
9 'モノクロの男性の肖像写真。軍服を着て石の階段に座っている。',
10 "庭で茶色の犬がこちらを向いて座っている。"
11]
12text_embeddings = model.encode(sentences)
13text_embeddings.shape
14# (2, 1024)
15
16image_urls = [
17 'https://upload.wikimedia.org/wikipedia/commons/7/73/Shigenobu_Okuma_5.jpg',
18 'https://upload.wikimedia.org/wikipedia/commons/7/78/Akita_inu.jpeg'
19]
20images = [
21 Image.open(io.BytesIO(requests.get(image_urls[0]).content)).resize((150, 240)),
22 Image.open(io.BytesIO(requests.get(image_urls[1]).content)).resize((240, 150))
23]
24
25image_embeddings = model.encode(images)
26image_embeddings.shape
27# (2, 1024)
28
29similarities = model.similarity(text_embeddings, image_embeddings)
30similarities
31# tensor([[ 2.6399e-01, 8.1531e-02],
32# [-2.4970e-04, 3.1410e-01]])