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