Skip to content

Commit 8b20a21

Browse files
committed
fix(inference): fix bug w padding @ slid win infer
1 parent 09faed3 commit 8b20a21

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cellseg_models_pytorch/inference/sliding_window_inferer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -124,8 +124,8 @@ def _get_margins(
124124
self, first_endpoint: int, img_size: int, stride: int, pad: int = None
125125
) -> Tuple[int, int]:
126126
"""Get the number of slices needed for one direction and the overlap."""
127-
if pad is not None:
128-
img_size += int(pad)
127+
pad = int(pad) if pad is not None else 20 # at least some padding needed
128+
img_size += pad
129129

130130
n = 1
131131
mod = 0

0 commit comments

Comments
 (0)