D3V1L1810 commited on
Commit
70d8c9d
·
verified ·
1 Parent(s): 448f141

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -36,7 +36,7 @@ def create_solutions(image_urls, all_bboxes, all_bboxes2, all_segments):
36
  box_id = 1
37
  cat_id = 1
38
  for image_url, bbox, bbox2, segmnt in zip(image_urls, all_bboxes, all_bboxes2, all_segments):
39
-
40
  for subbox, subbox2, subsegmnt in zip(bbox, bbox2, segmnt):
41
  w = subbox2[2]
42
  h = subbox2[3]
@@ -45,10 +45,11 @@ def create_solutions(image_urls, all_bboxes, all_bboxes2, all_segments):
45
  flattened_segmnt = [item for sublist in subsegmnt for item in sublist]
46
 
47
  ans = {"image_id": img_id, "id": box_id, "area": area, "category_id": cat_id, "bbox": subbox, "segment": flattened_segmnt}
48
- obj ={"url": image_url, "answer":[ans]}
49
- solutions.append(obj)
50
  box_id += 1
 
51
  img_id += 1
 
 
52
  return solutions
53
 
54
 
 
36
  box_id = 1
37
  cat_id = 1
38
  for image_url, bbox, bbox2, segmnt in zip(image_urls, all_bboxes, all_bboxes2, all_segments):
39
+ temp=[]
40
  for subbox, subbox2, subsegmnt in zip(bbox, bbox2, segmnt):
41
  w = subbox2[2]
42
  h = subbox2[3]
 
45
  flattened_segmnt = [item for sublist in subsegmnt for item in sublist]
46
 
47
  ans = {"image_id": img_id, "id": box_id, "area": area, "category_id": cat_id, "bbox": subbox, "segment": flattened_segmnt}
 
 
48
  box_id += 1
49
+ temp.append(ans)
50
  img_id += 1
51
+ obj ={"url": image_url, "answer":temp}
52
+ solutions.append(obj)
53
  return solutions
54
 
55