Update test_utils.py

Dw1010 · Mar 23, 2023 · 5bb0775 · 5bb0775
1 parent 6d23147
commit 5bb0775
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/core/test_utils.py b/core/test_utils.py
@@ -92,6 +92,7 @@ def render(self, params, bg_img=None):
         if bg_img is None:
             bg_img = np.ones_like(img_out) * 255
         else:
+            bg_img = imgUtils.pad2squre(bg_img)
             bg_img = cv.resize(bg_img, (self.render_size, self.render_size))
 
         img_out = img_out * mask_out + bg_img * (1 - mask_out)