Spaces:
Build error
Build error
Update app.py
Browse files
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 |
|