Update app.py
Browse files
app.py
CHANGED
@@ -33,6 +33,8 @@ def get_polygon(points, image, predictor):
|
|
33 |
"""
|
34 |
points = list(chain.from_iterable(points))
|
35 |
|
|
|
|
|
36 |
masks, _, _ = predictor.predict(
|
37 |
box=input_box[None, :],
|
38 |
multimask_output=False,
|
|
|
33 |
"""
|
34 |
points = list(chain.from_iterable(points))
|
35 |
|
36 |
+
input_box = np.array(points)
|
37 |
+
|
38 |
masks, _, _ = predictor.predict(
|
39 |
box=input_box[None, :],
|
40 |
multimask_output=False,
|