From dd346c3d58138d025fc9ae1ffb2030ebe5c4888e Mon Sep 17 00:00:00 2001 From: Luis Quesada Date: Wed, 15 May 2024 22:04:47 +0200 Subject: [PATCH] Add assert on stitch size vs batch size --- inpaint_cropandstitch.py | 1 + 1 file changed, 1 insertion(+) diff --git a/inpaint_cropandstitch.py b/inpaint_cropandstitch.py index c15082c..c887608 100644 --- a/inpaint_cropandstitch.py +++ b/inpaint_cropandstitch.py @@ -366,6 +366,7 @@ class InpaintStitch: results = [] batch_size = inpainted_image.shape[0] + assert len(stitch['x']) == batch_size, "Stitch size doesn't match image batch size" for b in range(batch_size): one_image = inpainted_image[b] one_stitch = {}