Spaces:
Runtime error
Runtime error
napatswift
commited on
Commit
•
ba7e7bf
1
Parent(s):
1e41e52
Change seq type
Browse files
main.py
CHANGED
@@ -107,7 +107,7 @@ def predict(image_input):
|
|
107 |
for mask_image in filtered_mask_images:
|
108 |
bbox_list.extend(get_bbox(mask_image.astype(np.uint8)))
|
109 |
|
110 |
-
return {'rect-fit': bbox_list, 'bbox': filtered_bboxes}
|
111 |
|
112 |
|
113 |
def run():
|
|
|
107 |
for mask_image in filtered_mask_images:
|
108 |
bbox_list.extend(get_bbox(mask_image.astype(np.uint8)))
|
109 |
|
110 |
+
return {'rect-fit': bbox_list, 'bbox': filtered_bboxes.tolist()}
|
111 |
|
112 |
|
113 |
def run():
|