From 5db1803430f02b9a2a6aa4ca36da06232b4d22e9 Mon Sep 17 00:00:00 2001 From: Gabriel Gutierrez Date: Mon, 11 Nov 2024 14:11:22 -0300 Subject: [PATCH] Update padding documentation in PatchInferencerEngine to clarify expected dimensions --- minerva/engines/patch_inferencer_engine.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/minerva/engines/patch_inferencer_engine.py b/minerva/engines/patch_inferencer_engine.py index 8fa8de3..eab7535 100644 --- a/minerva/engines/patch_inferencer_engine.py +++ b/minerva/engines/patch_inferencer_engine.py @@ -143,7 +143,7 @@ def __init__( Expected shape of the model output per patch. Defaults to input_shape. padding : dict, optional Padding configuration with keys: - - 'pad': Tuple of padding for each dimension, e.g., (0, 3, 3). + - 'pad': Tuple of padding for each expected final dimension, e.g., (0, 512, 512) - (c, h, w). - 'mode': Padding mode, e.g., 'constant', 'reflect'. - 'value': Padding value if mode is 'constant'. """