From a06035934366877ac6419e29bdebc93b07e0668c Mon Sep 17 00:00:00 2001
From: ShangGaoG <12132332@mail.sustech.edu.cn>
Date: Wed, 12 Apr 2023 19:05:11 +0800
Subject: [PATCH] note

---
 tools/base_segmenter.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tools/base_segmenter.py b/tools/base_segmenter.py
index b94f9ba..42e017c 100644
--- a/tools/base_segmenter.py
+++ b/tools/base_segmenter.py
@@ -30,7 +30,7 @@ class BaseSegmenter:
     @torch.no_grad()
     def set_image(self, image: np.ndarray):
         # PIL.open(image_path) 3channel: RGB
-        # image embedding
+        # image embedding: avoid encode the same image multiple times
         if self.embedded:
             print('repeat embedding, please reset_image.')
             return
@@ -40,7 +40,7 @@ class BaseSegmenter:
     
     @torch.no_grad()
     def reset_image(self):
-        # reset
+        # reset image embeding
         self.predictor.reset_image()
         self.embedded = False
 
@@ -71,7 +71,7 @@ class BaseSegmenter:
                                 mask_input=prompts['mask_input'], 
                                 multimask_output=multimask)
         else:
-            raise("Not implement!")
+            raise("Not implement now!")
         # masks (n, h, w), scores (n,), logits (n, 256, 256)
         return masks, scores, logits