Spaces:
Running
Running
Commit
·
5f60f25
1
Parent(s):
c53d901
app deploy
Browse files
app.py
CHANGED
@@ -102,7 +102,7 @@ def show_preds_image(image_path):
|
|
102 |
width=img_res_toshow.shape[1]
|
103 |
dim=(width,height)
|
104 |
|
105 |
-
outputs = model.predict(source=
|
106 |
|
107 |
#obtain BW image
|
108 |
bw=(outputs[0].masks.masks[0].cpu().numpy() * 255).astype("uint8")
|
@@ -130,7 +130,7 @@ def show_preds_image(image_path):
|
|
130 |
img,
|
131 |
(int(det[0]), int(det[1])),
|
132 |
(int(det[2]), int(det[3])),
|
133 |
-
color=(
|
134 |
thickness=2,
|
135 |
lineType=cv2.LINE_AA
|
136 |
)
|
|
|
102 |
width=img_res_toshow.shape[1]
|
103 |
dim=(width,height)
|
104 |
|
105 |
+
outputs = model.predict(source=img_res_toshow)
|
106 |
|
107 |
#obtain BW image
|
108 |
bw=(outputs[0].masks.masks[0].cpu().numpy() * 255).astype("uint8")
|
|
|
130 |
img,
|
131 |
(int(det[0]), int(det[1])),
|
132 |
(int(det[2]), int(det[3])),
|
133 |
+
color=(255, 0, 0),
|
134 |
thickness=2,
|
135 |
lineType=cv2.LINE_AA
|
136 |
)
|