Add local image embedding impls

This commit is contained in:
liyaxing
2023-09-26 20:13:55 +08:00
committed by tsukasa
parent 5f346b1bd9
commit 451ab5e0a8
5 changed files with 182 additions and 13 deletions
+9
View File
@@ -74,6 +74,15 @@ class TestVectorSTorage(unittest.TestCase):
image_data = KnowledgeStore().get_chunk_reader().read_chunk_list_to_single_bytes(image_obj.get_chunk_list())
self.assertEqual(file_size, len(image_data))
from PIL import Image
import io
image = Image.open(io.BytesIO(image_data))
image.show()
from sentence_transformers import SentenceTransformer
#model = SentenceTransformer('clip-ViT-B-32-multilingual-v1')
model = SentenceTransformer('clip-ViT-B-32')
model.encode(image, convert_to_tensor=True)
def test_relation(self):
obj1 = ObjectID.hash_data("12345".encode("utf-8"))