wzzanthony7 commited on
Commit
3d0fba7
·
verified ·
1 Parent(s): 506c0c1
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def RoboFlowGetOutlineBoxesPIL(pil_img):
30
  def localOutlineBox(image):
31
  model_name = "vt_dataset_yolov12_v7_weights.pt"
32
  model = YOLO(model_path)
33
- yolo_ret = model(image_path, verbose=False)
34
  useful_ret = yolo_ret[0]
35
  names = model.names
36
  all_box_info = []
@@ -44,6 +44,7 @@ def localOutlineBox(image):
44
  box_info['width'] = w
45
  box_info['height'] = h
46
  all_box_info.append(box_info)
 
47
  return all_box_info
48
  def calculate_iou(box1, box2):
49
  """计算两个框的IoU
 
30
  def localOutlineBox(image):
31
  model_name = "vt_dataset_yolov12_v7_weights.pt"
32
  model = YOLO(model_path)
33
+ yolo_ret = model(image, verbose=False)
34
  useful_ret = yolo_ret[0]
35
  names = model.names
36
  all_box_info = []
 
44
  box_info['width'] = w
45
  box_info['height'] = h
46
  all_box_info.append(box_info)
47
+ print(f"total length of all box_info is: {len(all_box_info)}")
48
  return all_box_info
49
  def calculate_iou(box1, box2):
50
  """计算两个框的IoU