Upload inference.py

inference.py

@@ -37,6 +37,9 @@ def inference(pipe, fiber_imgs, ring_imgs, num_steps):
     # sizes
     tile = 512
     canvas_size = tile * 2 
+    ratio = 45/145
+    original_width, original_height = (512,512)
+    new_width = int(original_width * ratio)
 
 
     # # stack channels: [fiber, ring, ring] -> H,W,3
@@ -75,11 +78,16 @@ def inference(pipe, fiber_imgs, ring_imgs, num_steps):
     # tr = pred.crop((tile, 0, canvas_size, tile))
     # bl = pred.crop((0, tile, tile, canvas_size))
     # br = pred.crop((tile, tile, canvas_size, canvas_size))
+    
+    ring_imgs = PIL.Image.fromarray(ring_imgs)  
+    # ring_imgs = PIL.Image.fromarray(ring_imgs)  
 
     tl = ring_imgs.crop((0, 0, tile, tile))
     tr = ring_imgs.crop((tile, 0, canvas_size, tile))
     bl = ring_imgs.crop((0, tile, tile, canvas_size))
     br = ring_imgs.crop((tile, tile, canvas_size, canvas_size))
+    
+    # resized_img_2 = cv2.resize(tr, (new_width, original_height), interpolation=cv2.INTER_LANCZOS4)
 
     # close opened images to free handles
     fiber_imgs.close()