Update app.py
Browse files
app.py
CHANGED
@@ -246,8 +246,12 @@ def run_grounded_sam(input_image, text_prompt, task_type, box_threshold, text_th
|
|
246 |
x_min, y_min, x_max, y_max = 250, 25, 774, 803
|
247 |
elif id == -1572157: #example 9 *
|
248 |
x_min, y_min, x_max, y_max = 15, 88, 1006, 977
|
|
|
|
|
|
|
|
|
249 |
else:
|
250 |
-
print("not defined")
|
251 |
raise NotImplementedError
|
252 |
bbox = np.array([x_min, y_min, x_max, y_max])
|
253 |
bbox = torch.tensor(bbox).unsqueeze(0)
|
@@ -502,7 +506,11 @@ if __name__ == "__main__":
|
|
502 |
with gr.Column():
|
503 |
gr.Examples(["example8.jpg"], inputs=input_image)
|
504 |
with gr.Column():
|
505 |
-
gr.Examples(["example9.jpg"], inputs=input_image)
|
|
|
|
|
|
|
|
|
506 |
run_button.click(fn=run_grounded_sam, inputs=[
|
507 |
input_image, text_prompt, task_type, box_threshold, text_threshold, iou_threshold], outputs=[gallery1,gallery2])
|
508 |
|
|
|
246 |
x_min, y_min, x_max, y_max = 250, 25, 774, 803
|
247 |
elif id == -1572157: #example 9 *
|
248 |
x_min, y_min, x_max, y_max = 15, 88, 1006, 977
|
249 |
+
elif id == -509470: #example 10
|
250 |
+
x_min, y_min, x_max, y_max = 190, 0, 530, 395
|
251 |
+
elif id == -42440: #example 11
|
252 |
+
x_min, y_min, x_max, y_max = 282, 134, 534, 394
|
253 |
else:
|
254 |
+
print(id, "not defined")
|
255 |
raise NotImplementedError
|
256 |
bbox = np.array([x_min, y_min, x_max, y_max])
|
257 |
bbox = torch.tensor(bbox).unsqueeze(0)
|
|
|
506 |
with gr.Column():
|
507 |
gr.Examples(["example8.jpg"], inputs=input_image)
|
508 |
with gr.Column():
|
509 |
+
gr.Examples(["example9.jpg"], inputs=input_image)
|
510 |
+
with gr.Column():
|
511 |
+
gr.Examples(["example10.jpg"], inputs=input_image)
|
512 |
+
with gr.Column():
|
513 |
+
gr.Examples(["example11.jpg"], inputs=input_image)
|
514 |
run_button.click(fn=run_grounded_sam, inputs=[
|
515 |
input_image, text_prompt, task_type, box_threshold, text_threshold, iou_threshold], outputs=[gallery1,gallery2])
|
516 |
|