fix: feadback.py

app_configs.py

@@ -2,4 +2,4 @@ model_type = r'vit_b'
 # model_ckpt_path = None
 model_ckpt_path = "checkpoints/sam_vit_b_01ec64.pth"
 device = None
-enable_segment_all = True
+enable_segment_all = False

feedback.py

@@ -17,11 +17,13 @@ def write_row(filepath:str, row: Dict):
 
 class Feedback():
     def __init__(self, 
-                 image_dir = './data/input', 
+                 image_dir = './data/image', 
                  mask_dir = './data/mask',
                  inference_csv = './data/inference.csv',
                  feedback_csv = './data/feedback.csv',
                  ):
+        os.makedirs(image_dir, exist_ok=True)
+        os.makedirs(mask_dir, exist_ok=True)
         self.image_dir = image_dir
         self.mask_dir = mask_dir
         self.inference_csv = inference_csv
@@ -29,12 +31,16 @@ class Feedback():
 
     def save_inference(self, pt_coords:List, pt_labels:List, image: Image.Image, mask: np.ndarray):
         self.inference_id = uuid.uuid4()
+        image_path = os.path.join(self.image_dir,f'{self.inference_id}.png')
+        mask_path = os.path.join(self.mask_dir, f'{self.inference_id}.npy')
+        image.save(image_path)
+        np.save(mask_path, mask)
         write_row(
             filepath=self.inference_csv,
             row = {
                 "inference_id": self.inference_id,
-                "image": image.tobytes(),
-                "mask": mask.tobytes(),
+                "image": image_path,
+                "mask": mask_path,
                 "pt_coords": str(pt_coords),
                 "pt_labels": str(pt_labels),
             }