fix(api): calculate outpaint overlap factor before adding margins

ssube · Feb 7, 2023 · b6fd288 · b6fd288
1 parent a9f4e19
commit b6fd288
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/api/onnx_web/chain/upscale_outpaint.py b/api/onnx_web/chain/upscale_outpaint.py
@@ -35,6 +35,10 @@ def upscale_outpaint(
     prompt = prompt or params.prompt
     logger.info("upscaling image by expanding borders: %s", border)
 
+    margin_x = float(max(border.left, border.right))
+    margin_y = float(max(border.top, border.bottom))
+    overlap = min(margin_x / source_image.width, margin_y / source_image.height)
+
     if mask_image is None:
         # if no mask was provided, keep the full source image
         mask_image = Image.new("RGB", source_image.size, "black")
@@ -108,10 +112,6 @@ def outpaint(image: Image.Image, dims: Tuple[int, int, int]):
         draw_mask.rectangle((left, top, left + tile, top + tile), fill="black")
         return result.images[0]
 
-    margin_x = float(max(border.left, border.right))
-    margin_y = float(max(border.top, border.bottom))
-    overlap = min(margin_x / source_image.width, margin_y / source_image.height)
-
     if overlap == 0:
         logger.debug("outpainting with 0 margin, using grid tiling")
         output = process_tile_grid(source_image, SizeChart.auto, 1, [outpaint])