From bbacca3c8042436778feddb0130b785942313c34 Mon Sep 17 00:00:00 2001 From: kirilspodgalskis Date: Wed, 27 Nov 2024 17:42:49 +0200 Subject: [PATCH 1/2] make resize argument value actually used for resizing screenshot --- cv_pom/frameworks/os_gui.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cv_pom/frameworks/os_gui.py b/cv_pom/frameworks/os_gui.py index 97f1539..7a6db93 100644 --- a/cv_pom/frameworks/os_gui.py +++ b/cv_pom/frameworks/os_gui.py @@ -28,7 +28,7 @@ def __init__(self, model_path: Path | str, **kwargs) -> None: def _get_screenshot(self) -> np.ndarray: screenshot = pyautogui.screenshot() width, height = screenshot.size - screenshot = screenshot.resize((width // 2, height // 2), Image.LANCZOS) + screenshot = screenshot.resize((width*self.resize, height*self.resize), Image.LANCZOS) pimg = np.array(screenshot) return cv.cvtColor(np.array(pimg), cv.COLOR_RGB2BGR) From b6f237a729469a8880468842828ae6e3c255d58b Mon Sep 17 00:00:00 2001 From: kirilspodgalskis Date: Fri, 29 Nov 2024 12:48:38 +0200 Subject: [PATCH 2/2] round pixels to whole number --- cv_pom/frameworks/os_gui.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cv_pom/frameworks/os_gui.py b/cv_pom/frameworks/os_gui.py index 7a6db93..df1f0d6 100644 --- a/cv_pom/frameworks/os_gui.py +++ b/cv_pom/frameworks/os_gui.py @@ -28,7 +28,7 @@ def __init__(self, model_path: Path | str, **kwargs) -> None: def _get_screenshot(self) -> np.ndarray: screenshot = pyautogui.screenshot() width, height = screenshot.size - screenshot = screenshot.resize((width*self.resize, height*self.resize), Image.LANCZOS) + screenshot = screenshot.resize((round(width*self.resize), round(height*self.resize)), Image.LANCZOS) pimg = np.array(screenshot) return cv.cvtColor(np.array(pimg), cv.COLOR_RGB2BGR)