Spaces:
Sleeping
Sleeping
Update python_utils/get_model.py
Browse files
python_utils/get_model.py
CHANGED
|
@@ -35,6 +35,7 @@ def load_model():
|
|
| 35 |
|
| 36 |
def mask_nms(masks, scores, nms_threshold=0.5):
|
| 37 |
import supervision as sv
|
|
|
|
| 38 |
|
| 39 |
polygons = []
|
| 40 |
for mask in masks:
|
|
|
|
| 35 |
|
| 36 |
def mask_nms(masks, scores, nms_threshold=0.5):
|
| 37 |
import supervision as sv
|
| 38 |
+
from shapely.geometry.polygon import Polygon
|
| 39 |
|
| 40 |
polygons = []
|
| 41 |
for mask in masks:
|