Fix cv2

app.py

@@ -9,7 +9,6 @@ from PIL import Image
 import numpy as np
 import torch
 import legacy
-import cv2
 import paddlehub as hub
 
 u2net = hub.Module(name='U2Net')
@@ -76,19 +75,18 @@ def inpaint(input_img, mask, option):
         mask = Image.fromarray(result[0]['mask'])
     else:
         mask = mask.resize((width,height))
+    
+    input_img = input_img.resize(512, 512)
+    mask = mask.resize(512, 512)
 
     mask = mask.convert('L')
     mask = np.array(mask) / 255.
-    mask = cv2.resize(mask,
-            (512, 512), interpolation=cv2.INTER_NEAREST)
     mask_tensor = torch.from_numpy(mask).to(torch.float32)
     mask_tensor = mask_tensor.unsqueeze(0)
     mask_tensor = mask_tensor.unsqueeze(0).to(device)
 
     rgb = input_img.convert('RGB')
     rgb = np.array(rgb)
-    rgb = cv2.resize(rgb,
-            (512, 512), interpolation=cv2.INTER_AREA)
     rgb = rgb.transpose(2,0,1)
     rgb = torch.from_numpy(rgb.astype(np.float32)).unsqueeze(0)
     rgb = (rgb.to(torch.float32) / 127.5 - 1).to(device)

requirements.txt

@@ -1,4 +1,3 @@
-opencv-python
 icecream
 psutil
 click