Commit
·
363ec41
1
Parent(s):
d1eaab8
Update app.py
Browse files
app.py
CHANGED
@@ -54,9 +54,9 @@ image_output = gr.outputs.Image(type="pil", label="Output Image")
|
|
54 |
|
55 |
|
56 |
def gen_pred(img, model=seg_model):
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
# # img = cv2.imread("./003e2c95d.jpg")
|
61 |
img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]
|
62 |
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|
|
|
54 |
|
55 |
|
56 |
def gen_pred(img, model=seg_model):
|
57 |
+
pil_image = img.convert('RGB')
|
58 |
+
open_cv_image = np.array(pil_image)
|
59 |
+
img = open_cv_image[:, :, ::-1].copy()
|
60 |
# # img = cv2.imread("./003e2c95d.jpg")
|
61 |
img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]
|
62 |
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|