From abf8296fc1ab7795a22441b7c29feb5bb902938b Mon Sep 17 00:00:00 2001
From: polarbean <harry.carey95@gmail.com>
Date: Mon, 17 Feb 2025 17:05:08 +0100
Subject: [PATCH] fix region padding based on gergelys feedback

---
 .vscode/settings.json                   | 11 ++++++++++-
 PyNutil/processing/counting_and_load.py |  5 ++++-
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/.vscode/settings.json b/.vscode/settings.json
index de288e1..d35c541 100644
--- a/.vscode/settings.json
+++ b/.vscode/settings.json
@@ -1,3 +1,12 @@
 {
-    "python.formatting.provider": "black"
+    "python.formatting.provider": "black",
+    "python.testing.unittestArgs": [
+        "-v",
+        "-s",
+        "./tests",
+        "-p",
+        "test*.py"
+    ],
+    "python.testing.pytestEnabled": false,
+    "python.testing.unittestEnabled": true
 }
\ No newline at end of file
diff --git a/PyNutil/processing/counting_and_load.py b/PyNutil/processing/counting_and_load.py
index 4fdcf32..f7498e2 100644
--- a/PyNutil/processing/counting_and_load.py
+++ b/PyNutil/processing/counting_and_load.py
@@ -209,7 +209,10 @@ def warp_image(image, triangulation, rescaleXY):
     newY[newY >= reg_h] = reg_h - 1
     newX[newX < 0] = 0
     newY[newY < 0] = 0
-    new_image = image[newY, newX]
+    new_image = np.zeros_like(image)
+    mask = (newX <= reg_w) & (newY <= reg_h) & (newX > 0) & (newY > 0)
+    new_image[mask] = image[mask]
+    new_image[~mask] = 0
     return new_image
 
 
-- 
GitLab