fix image padding

main.py

@@ -173,6 +173,7 @@ def load_models():
 #load_models()
 
 def img_pad(img, tw, th, transparent=False):
+    #print('pad', tw, th)
     img.thumbnail((tw, th))
     if transparent:
         new_img = Image.new('RGBA', (tw, th), (0, 0, 0, 0))
@@ -180,12 +181,15 @@ def img_pad(img, tw, th, transparent=False):
         new_img = Image.new("RGB", (tw, th), (0, 0, 0))
     left = (tw - img.width) // 2
     top = (th - img.height) // 2
+    #print(left, top)
     new_img.paste(img, (left, top))
     return new_img
 
 
 def resize_pad(img, tw, th, transparent):
     w, h = img.size
+    orig_tw = tw
+    orig_th = th
     
     if tw/th > w/h:
         tw = int(th * w/h)
@@ -194,7 +198,7 @@ def resize_pad(img, tw, th, transparent):
     
     img = img.resize((tw, th), Image.BICUBIC)
 
-    return img_pad(img, tw, th, transparent)
+    return img_pad(img, orig_tw, orig_th, True)
 
 
 def resize_and_pad(img, target_img):