remove cv

gradio_painttransformer.py

@@ -16,7 +16,6 @@ import gradio as gr
 
 import paddle
 import render_utils
-import render_parallel
 import render_serial
 
 # ---------- Settings ----------

render_serial.py

@@ -8,7 +8,7 @@ import paddle.nn.functional as F
 import numpy as np
 from PIL import Image
 import math
-import cv2
+# import cv2
 import render_utils
 import time
 

render_utils.py

@@ -82,11 +82,11 @@ def read_img(img_path, img_type='RGB', h=None, w=None):
     img = paddle.to_tensor(img).unsqueeze(0).astype('float32') / 255.
     return img
 
-def preprocess(img, w=512, h=512):
-    image = cv2.resize(img, (w, h), cv2.INTER_NEAREST)
-    image = image.transpose((2, 0, 1))
-    image = paddle.to_tensor(image).unsqueeze(0).astype('float32') / 255.
-    return image
+# def preprocess(img, w=512, h=512):
+#     image = cv2.resize(img, (w, h), cv2.INTER_NEAREST)
+#     image = image.transpose((2, 0, 1))
+#     image = paddle.to_tensor(image).unsqueeze(0).astype('float32') / 255.
+#     return image
 
 def pad(img, H, W):
     b, c, h, w = img.shape