Commit
·
2d24604
1
Parent(s):
88d04c7
Update app.py
Browse files
app.py
CHANGED
@@ -64,8 +64,6 @@ def gen_pred(img, model=seg_model):
|
|
64 |
img = tf.expand_dims(img, axis=0)
|
65 |
pred = model.predict(img)
|
66 |
pred = np.squeeze(pred, axis=0)
|
67 |
-
print(len(pred))
|
68 |
-
print(max(pred))
|
69 |
img_bytes = pred.tobytes()
|
70 |
nparr = np.frombuffer(img_bytes, np.byte)
|
71 |
pred_pil = cv2.imdecode(nparr, cv2.IMREAD_REDUCED_COLOR_8)
|
|
|
64 |
img = tf.expand_dims(img, axis=0)
|
65 |
pred = model.predict(img)
|
66 |
pred = np.squeeze(pred, axis=0)
|
|
|
|
|
67 |
img_bytes = pred.tobytes()
|
68 |
nparr = np.frombuffer(img_bytes, np.byte)
|
69 |
pred_pil = cv2.imdecode(nparr, cv2.IMREAD_REDUCED_COLOR_8)
|