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