Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,6 @@ import cv2
|
|
2 |
import numpy as np
|
3 |
import gradio as gr
|
4 |
import requests
|
5 |
-
from io import BytesIO
|
6 |
|
7 |
# Configuration files
|
8 |
config_file = "ssd_mobilenet_v3_large_coco_2020_01_14.pbtxt"
|
@@ -48,11 +47,7 @@ def detect_objects(frame):
|
|
48 |
}
|
49 |
detections.append(detected_object)
|
50 |
|
51 |
-
|
52 |
-
cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0), 2)
|
53 |
-
cv2.putText(frame, classLabels[ClassInd - 1].upper(), (x, y-10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
|
54 |
-
|
55 |
-
return frame, detections
|
56 |
|
57 |
def get_image_from_url(url):
|
58 |
response = requests.get(url)
|
@@ -62,16 +57,16 @@ def get_image_from_url(url):
|
|
62 |
|
63 |
def detect_objects_in_image_url(url):
|
64 |
frame = get_image_from_url(url)
|
65 |
-
|
66 |
-
return
|
67 |
|
68 |
# Define the Gradio interface
|
69 |
iface = gr.Interface(
|
70 |
fn=detect_objects_in_image_url,
|
71 |
inputs="text",
|
72 |
-
outputs=
|
73 |
title="Object Detection",
|
74 |
-
description="Enter an image URL to detect objects. The
|
75 |
)
|
76 |
|
77 |
# Launch the interface
|
|
|
2 |
import numpy as np
|
3 |
import gradio as gr
|
4 |
import requests
|
|
|
5 |
|
6 |
# Configuration files
|
7 |
config_file = "ssd_mobilenet_v3_large_coco_2020_01_14.pbtxt"
|
|
|
47 |
}
|
48 |
detections.append(detected_object)
|
49 |
|
50 |
+
return detections
|
|
|
|
|
|
|
|
|
51 |
|
52 |
def get_image_from_url(url):
|
53 |
response = requests.get(url)
|
|
|
57 |
|
58 |
def detect_objects_in_image_url(url):
|
59 |
frame = get_image_from_url(url)
|
60 |
+
detected_objects = detect_objects(frame)
|
61 |
+
return detected_objects
|
62 |
|
63 |
# Define the Gradio interface
|
64 |
iface = gr.Interface(
|
65 |
fn=detect_objects_in_image_url,
|
66 |
inputs="text",
|
67 |
+
outputs=gr.JSON(),
|
68 |
title="Object Detection",
|
69 |
+
description="Enter an image URL to detect objects. The detected objects will be returned as JSON."
|
70 |
)
|
71 |
|
72 |
# Launch the interface
|