CloudAnts commited on
Commit
187268a
·
1 Parent(s): eeff366
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -89,15 +89,15 @@ def process_image():
89
  bounding_box_annotator = sv.BoundingBoxAnnotator()
90
  label_annotator = sv.LabelAnnotator()
91
  image_dir = "./data1"
92
- files = os.listdir('./data1')
93
  files.sort()
94
  files = files[0:100]
95
  print(files)
96
  counter = 0
97
  for ii in files:
98
- random_image_data = cv2.imread('./data1/' + ii)
99
- random_image_data1 = cv2.imread('./data1/' + ii)
100
- results = model(source='./data1/' + ii, conf=0.07)[0]
101
  detections = sv.Detections.from_ultralytics(results)
102
  annotated_image = bounding_box_annotator.annotate(scene=random_image_data, detections=detections)
103
  annotated_image = label_annotator.annotate(scene=annotated_image, detections=detections)
 
89
  bounding_box_annotator = sv.BoundingBoxAnnotator()
90
  label_annotator = sv.LabelAnnotator()
91
  image_dir = "./data1"
92
+ files = os.listdir('./mnt/data1')
93
  files.sort()
94
  files = files[0:100]
95
  print(files)
96
  counter = 0
97
  for ii in files:
98
+ random_image_data = cv2.imread('./mnt/data1/' + ii)
99
+ random_image_data1 = cv2.imread('./mnt/data1/' + ii)
100
+ results = model(source='./mnt/data1/' + ii, conf=0.07)[0]
101
  detections = sv.Detections.from_ultralytics(results)
102
  annotated_image = bounding_box_annotator.annotate(scene=random_image_data, detections=detections)
103
  annotated_image = label_annotator.annotate(scene=annotated_image, detections=detections)