fix final cropping

styleTransfer.py

@@ -6,19 +6,24 @@ import cv2
 
 def resize_sofa(img):
     img = Image.fromarray(img)
-    width, height = img.size
+    width, height = img.size 
     idx = np.argmin([width,height])
     newsize = (640, 640) # parameters from test script
 
     if idx==0:
         img1 = Image.new(img.mode, (height, height), (255, 255, 255))
         img1.paste(img, ((height-width)//2, 0))
-        box = ((height-width)//2*newsize[0]/width, 0, newsize[0] - (height-width)//2*newsize[0]/width, newsize[1])
+        box = ( newsize[0]*(1-width/height)//2,
+                0, 
+                newsize[0]-newsize[0]*(1-width/height)//2,
+                newsize[1])
     else:
         img1 = Image.new(img.mode, (width, width), (255, 255, 255))
-        img1.paste(img, (0, (width-height)//2))
-        box = (0, (height-width)//2*newsize[1]/height, newsize[0], newsize[1]-(height-width)//2*newsize[1]/height)
-
+        img1.paste(img, (0, (width-height)//2)) 
+        box = (0, 
+                newsize[1]*(1-height/width)//2, 
+                newsize[0], 
+                newsize[1]-newsize[1]*(1-height/width)//2)
     im1 = img1.resize(newsize)
     return im1,box