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,
|