bugfix: patch writer - remove hard coded values

deepfakes · Sep 30, 2023 · 3336772 · 3336772
1 parent f23c87c
commit 3336772
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/plugins/convert/writer/patch.py b/plugins/convert/writer/patch.py
@@ -45,7 +45,8 @@ def __init__(self, output_folder: str, patch_size: int, **kwargs) -> None:
 
         self._dummy_patch = np.zeros((1, patch_size, patch_size, 4), dtype=np.float32)
 
-        tl_box = np.array([[0, 0], [128, 0], [128, 128], [0, 128]], dtype=np.float32)
+        tl_box = np.array([[0, 0], [patch_size, 0], [patch_size, patch_size], [0, patch_size]],
+                          dtype=np.float32)
         self._patch_corner = {"top-left": tl_box[0],
                               "top-right": tl_box[1],
                               "bottom-right": tl_box[2],