solve image_change

2025-12-16 16:37:58 +01:00 · 2023-04-14 17:43:05 +08:00
parent cddb447e3f
commit 0000a2a103
2 changed files with 4 additions and 2 deletions
--- a/tools/base_segmenter.py
+++ b/tools/base_segmenter.py
@@ -31,6 +31,7 @@ class BaseSegmenter:
    def set_image(self, image: np.ndarray):
        # PIL.open(image_path) 3channel: RGB
        # image embedding: avoid encode the same image multiple times
+        self.orignal_image = image
        if self.embedded:
            print('repeat embedding, please reset_image.')
            return