jatamura commited on
Commit
27af8e6
·
verified ·
1 Parent(s): 5ff6df0

Update python_utils/get_model.py

Browse files
Files changed (1) hide show
  1. python_utils/get_model.py +1 -0
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: