Commit
·
8b7ee47
1
Parent(s):
67680bf
Update app.py
Browse files
app.py
CHANGED
@@ -58,10 +58,10 @@ rows = 1
|
|
58 |
columns = 1
|
59 |
|
60 |
def gen_pred(img, model=seg_model):
|
61 |
-
|
62 |
-
|
63 |
-
|
64 |
-
img = cv2.imread("./003e2c95d.jpg")
|
65 |
img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]
|
66 |
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|
67 |
img = img/255
|
|
|
58 |
columns = 1
|
59 |
|
60 |
def gen_pred(img, model=seg_model):
|
61 |
+
pil_image = img.convert('RGB')
|
62 |
+
open_cv_image = np.array(pil_image)
|
63 |
+
img = open_cv_image[:, :, ::-1].copy()
|
64 |
+
# img = cv2.imread("./003e2c95d.jpg")
|
65 |
img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]
|
66 |
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|
67 |
img = img/255
|