From fb81ea509fa32ff28e6be3d1ba713405a4223523 Mon Sep 17 00:00:00 2001
From: Radionic <khpluslt@gmail.com>
Date: Fri, 29 Sep 2023 16:14:41 +0800
Subject: [PATCH] fix: image metadata not saved

---
 sam/sam.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/sam/sam.py b/sam/sam.py
index 7c03ffd..deafdfa 100644
--- a/sam/sam.py
+++ b/sam/sam.py
@@ -60,6 +60,13 @@ class SAM:
             predictor.set_image(image_np)
             emb = predictor.get_image_embedding().cpu().numpy()
             np.save(emb_filename, emb)
+
+            with open(f"{self.output_dir}/{embedding_id}.json", "w") as f:
+                data = {
+                    "input_size": predictor.input_size,
+                    "original_size": predictor.original_size,
+                }
+                json.dump(data, f)
         else:
             emb = np.load(emb_filename)