Commit
·
524bce7
1
Parent(s):
d9d4718
Update app.py
Browse files
app.py
CHANGED
@@ -61,7 +61,7 @@ def gen_pred(img=inputs, model=seg_model):
|
|
61 |
# img = cv2.imread(rgb_path)
|
62 |
# img = cv2.imread("./003e2c95d.jpg")
|
63 |
pil_image = Image.open('./003b50a15.jpg')
|
64 |
-
img = cv2.cvtColor(
|
65 |
img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]
|
66 |
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|
67 |
img = img/255
|
|
|
61 |
# img = cv2.imread(rgb_path)
|
62 |
# img = cv2.imread("./003e2c95d.jpg")
|
63 |
pil_image = Image.open('./003b50a15.jpg')
|
64 |
+
img = cv2.cvtColor(np.array(pil_image), cv2.COLOR_RGB2BGR)
|
65 |
img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]
|
66 |
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|
67 |
img = img/255
|