From 8923be74f516c9d475dee76301efedfb2973d4c0 Mon Sep 17 00:00:00 2001
From: nickroh <48642955+nickroh@users.noreply.github.com>
Date: Sat, 19 Oct 2024 17:35:35 -0500
Subject: [PATCH 1/2] Created Export & Import for img embedding

---
 sam2/sam2_image_predictor.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/sam2/sam2_image_predictor.py b/sam2/sam2_image_predictor.py
index 41ce53af5..a69974d3d 100644
--- a/sam2/sam2_image_predictor.py
+++ b/sam2/sam2_image_predictor.py
@@ -456,6 +456,22 @@ def get_image_embedding(self) -> torch.Tensor:
     def device(self) -> torch.device:
         return self.model.device
 
+    def import_fields(self, fields):
+        self._features = fields["features"]
+        self._orig_hw = fields["_orig_hw"]
+        self._is_image_set = True
+        
+    def export_fields(self):
+        """
+        Exports the specified fields of the TestSamPredictor class.
+        Returns:
+            dict: A dictionary containing the values of features, orig_h, orig_w, input_h, and input_w.
+        """
+        return {
+            "_features": self._features,
+            "_orig_hw": self._orig_hw
+        }
+    
     def reset_predictor(self) -> None:
         """
         Resets the image embeddings and other state variables.

From 4ff3bb163c662a4c92364bda44c4c9658f684ca1 Mon Sep 17 00:00:00 2001
From: nickroh <48642955+nickroh@users.noreply.github.com>
Date: Sun, 1 Dec 2024 18:25:16 -0600
Subject: [PATCH 2/2] Update sam2_image_predictor.py

---
 sam2/sam2_image_predictor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sam2/sam2_image_predictor.py b/sam2/sam2_image_predictor.py
index a69974d3d..2cd35384d 100644
--- a/sam2/sam2_image_predictor.py
+++ b/sam2/sam2_image_predictor.py
@@ -457,7 +457,7 @@ def device(self) -> torch.device:
         return self.model.device
 
     def import_fields(self, fields):
-        self._features = fields["features"]
+        self._features = fields["_features"]
         self._orig_hw = fields["_orig_hw"]
         self._is_image_set = True