remove unnecessary lines of code

OpenAdaptAI · Jun 5, 2023 · ad39a80 · ad39a80
1 parent 8ea9773
commit ad39a80
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/openadapt/strategies/demo.py b/openadapt/strategies/demo.py
@@ -44,10 +44,10 @@ def get_next_action_event(
         screenshot: Screenshot,
         window_event: WindowEvent,
     ):
-        #ascii_text = self.get_ascii_text(screenshot)
+        ascii_text = self.get_ascii_text(screenshot)
         #logger.info(f"ascii_text=\n{ascii_text}")
 
-        #ocr_text = self.get_ocr_text(screenshot)
+        ocr_text = self.get_ocr_text(screenshot)
         #logger.info(f"ocr_text=\n{ocr_text}")
 
         screenshot_bbox = self.get_screenshot_bbox(screenshot)

diff --git a/openadapt/strategies/mixins/sam.py b/openadapt/strategies/mixins/sam.py
@@ -73,8 +73,6 @@ def get_screenshot_bbox(self, screenshot: Screenshot, show_plots=SHOW_PLOTS) ->
         array_resized = np.array(image_resized)
         masks = self.sam_mask_generator.generate(array_resized)
         bbox_list = []
-        plt.figure(figsize=(10, 10))
-        plt.imshow(array_resized)
         for mask in masks:
             bbox_list.append(mask["bbox"])
         if SHOW_PLOTS :
@@ -122,8 +120,10 @@ def get_click_event_bbox(self, screenshot: Screenshot, show_plots=SHOW_PLOTS) ->
                 # Calculate bounding box coordinates
                 x0 = np.min(cols)
                 y0 = np.min(rows)
-                w = np.max(cols)
-                h = np.max(rows)
+                x1 = np.max(cols)
+                y1 = np.max(rows)
+                w = x1-x0
+                h = y1-y0
                 input_box = [x0, y0, w, h]
                 if SHOW_PLOTS :
                     plt.figure(figsize=(10, 10))
@@ -133,7 +133,7 @@ def get_click_event_bbox(self, screenshot: Screenshot, show_plots=SHOW_PLOTS) ->
                     show_points(input_point, input_label, plt.gca())
                     plt.axis("on")
                     plt.show()
-                return [x0, y0, w - x0, h - y0]
+                return input_box
         return []