Shengxiao0709 commited on
Commit
dd17bac
·
verified ·
1 Parent(s): a5f49f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -68,6 +68,7 @@ def parse_first_bbox(bboxes):
68
  return None
69
 
70
  def segment(annot_value):
 
71
  """
72
  annot_value 形如 [image_path, bboxes]
73
  - image_path: 字符串
 
68
  return None
69
 
70
  def segment(annot_value):
71
+ prepare_cuda()
72
  """
73
  annot_value 形如 [image_path, bboxes]
74
  - image_path: 字符串