qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-08-09 04:12:42 +02:00
Code Issues Projects Releases Wiki Activity

tiling: avoid self-overlapping windows if margin too wide

Browse source
This commit is contained in:
Robert Sachunsky 2026-07-30 15:29:42 +02:00
parent 5f4e4fb8bd
commit 628e76cae7
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/eynollah/utils/tiling.py
View file

@ -63,6 +63,10 @@ def do_prediction(
logger.debug("Patch size: %sx%s", img_height_model, img_width_model) logger.debug("Patch size: %sx%s", img_height_model, img_width_model)
margin = int(marginal_of_patch_percent * img_height_model) margin = int(marginal_of_patch_percent * img_height_model)
if 2 * margin > 0.5 * img_width_model:
margin = img_width_model // 4
if 2 * margin > 0.5 * img_height_model:
margin = img_height_model // 4
window = 1 / (1 + np.exp(5.0 - 5 * np.arange(2 * margin) / margin)) window = 1 / (1 + np.exp(5.0 - 5 * np.arange(2 * margin) / margin))
width_mid = img_width_model - 2 * margin width_mid = img_width_model - 2 * margin
height_mid = img_height_model - 2 * margin height_mid = img_height_model - 2 * margin
@ -245,6 +249,10 @@ def do_prediction_new_concept(
logger.debug("Patch size: %sx%s", img_height_model, img_width_model) logger.debug("Patch size: %sx%s", img_height_model, img_width_model)
margin = int(marginal_of_patch_percent * img_height_model) margin = int(marginal_of_patch_percent * img_height_model)
if 2 * margin > 0.5 * img_width_model:
margin = img_width_model // 4
if 2 * margin > 0.5 * img_height_model:
margin = img_height_model // 4
window = 1 / (1 + np.exp(5.0 - 5 * np.arange(2 * margin) / margin)) window = 1 / (1 + np.exp(5.0 - 5 * np.arange(2 * margin) / margin))
width_mid = img_width_model - 2 * margin width_mid = img_width_model - 2 * margin
height_mid = img_height_model - 2 * margin height_mid = img_height_model - 2 * margin