Duplicate from radames/segment-anything-embeddings-base

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,8 @@
+---
+license: apache-2.0
+tags:
+- feature-extraction
+library_name: generic
+pipeline_tag: image-classification
+duplicated_from: radames/segment-anything-embeddings-base
+---

pipeline.py

@@ -0,0 +1,24 @@
+from typing import List
+import torch
+from transformers import SamModel, SamProcessor
+from PIL import Image
+import numpy as np
+
+class PreTrainedPipeline():
+    def __init__(self, path=""):
+
+        self.device = torch.device(
+            "cuda" if torch.cuda.is_available() else "cpu")
+        self.processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
+        self.model = SamModel.from_pretrained(
+            "facebook/sam-vit-base").to(self.device)
+        self.model.eval()
+        self.model = self.model.to(self.device)
+
+    def __call__(self, inputs: "Image.Image") -> List[float]:
+        raw_image = inputs.convert("RGB")
+        inputs = self.processor(raw_image, return_tensors="pt").to(self.device)
+        feature_vector = self.model.get_image_embeddings(
+            inputs["pixel_values"])
+
+        return feature_vector.tolist()

requirements.txt

@@ -0,0 +1,4 @@
+git+https://github.com/huggingface/transformers.git
+torch
+numpy
+Pillow