fix(inference): fix padding bug.

okunator · okunator · commit 295ec77c2f6f · 2022-09-09T14:15:34.000+03:00
diff --git a/cellseg_models_pytorch/inference/sliding_window_inferer.py b/cellseg_models_pytorch/inference/sliding_window_inferer.py
@@ -161,8 +161,11 @@ def _infer_batch(self, input_batch: torch.Tensor) -> Dict[str, torch.Tensor]:
             self.stride, self.patch_size, tuple(input_batch.shape[2:]), self.padding
         )
 
-        padx = padx // 2
-        pady = pady // 2
+        padx, modx = divmod(padx, 2)
+        pady, mody = divmod(pady, 2)
+        padx += modx
+        pady += mody
+
         input_batch = F.pad(
             input_batch.float(), pad=(padx, padx, pady, pady), mode="reflect"
         )

Original file line number	Diff line number	Diff line change
`@@ -161,8 +161,11 @@ def _infer_batch(self, input_batch: torch.Tensor) -> Dict[str, torch.Tensor]:`
`161`	`161`	`self.stride, self.patch_size, tuple(input_batch.shape[2:]), self.padding`
`162`	`162`	`)`
`163`	`163`
`164`		`- padx = padx // 2`
`165`		`- pady = pady // 2`
	`164`	`+ padx, modx = divmod(padx, 2)`
	`165`	`+ pady, mody = divmod(pady, 2)`
	`166`	`+ padx += modx`
	`167`	`+ pady += mody`
	`168`	`+`
`166`	`169`	`input_batch = F.pad(`
`167`	`170`	`input_batch.float(), pad=(padx, padx, pady, pady), mode="reflect"`
`168`	`171`	`)`