Update app.py
Browse files
app.py
CHANGED
@@ -273,7 +273,7 @@ def object_detection_density_edge(image, conf_threshold=0.25, iou_threshold=0.45
|
|
273 |
|
274 |
|
275 |
# Function to resize and encode an image
|
276 |
-
def resize_image(image, max_size=
|
277 |
print("resize_image initialized")
|
278 |
if max(image.size) > max_size:
|
279 |
|
@@ -300,7 +300,7 @@ def resize_image(image, max_size=2500):
|
|
300 |
return base64.b64encode(buffered).decode("utf-8")
|
301 |
|
302 |
|
303 |
-
def infer(image, model, version, api_key, confidence=0.
|
304 |
|
305 |
base_url = f"https://detect.roboflow.com/{model}/{version}?api_key={api_key}&confidence={confidence}&overlap={overlap}&format={format}"
|
306 |
|
@@ -343,11 +343,11 @@ interface = gr.Interface(
|
|
343 |
gr.Textbox(value="sku-110k", label="Model Name"),
|
344 |
gr.Textbox(value="2", label="Model Version"),
|
345 |
gr.Textbox(value="gHiUgOSq9GqTnRy5mErk", label="API Key"),
|
346 |
-
gr.Slider(0.0, 1.0, value=0.
|
347 |
-
gr.Slider(0.0, 1.0, value=0.
|
348 |
gr.Radio(["json", "image"], value="image", label="Output Format"),
|
349 |
gr.Checkbox(False, label="Include Labels"),
|
350 |
-
gr.Slider(1, 10, value=
|
351 |
],
|
352 |
outputs=[
|
353 |
gr.Image(label="Roboflow Output Image"),
|
|
|
273 |
|
274 |
|
275 |
# Function to resize and encode an image
|
276 |
+
def resize_image(image, max_size=1500):
|
277 |
print("resize_image initialized")
|
278 |
if max(image.size) > max_size:
|
279 |
|
|
|
300 |
return base64.b64encode(buffered).decode("utf-8")
|
301 |
|
302 |
|
303 |
+
def infer(image, model, version, api_key, confidence=0.4, overlap=0.20, format="json", labels=False, stroke=1):
|
304 |
|
305 |
base_url = f"https://detect.roboflow.com/{model}/{version}?api_key={api_key}&confidence={confidence}&overlap={overlap}&format={format}"
|
306 |
|
|
|
343 |
gr.Textbox(value="sku-110k", label="Model Name"),
|
344 |
gr.Textbox(value="2", label="Model Version"),
|
345 |
gr.Textbox(value="gHiUgOSq9GqTnRy5mErk", label="API Key"),
|
346 |
+
gr.Slider(0.0, 1.0, value=0.40, label="Confidence Threshold"),
|
347 |
+
gr.Slider(0.0, 1.0, value=0.20, label="Overlap Threshold"),
|
348 |
gr.Radio(["json", "image"], value="image", label="Output Format"),
|
349 |
gr.Checkbox(False, label="Include Labels"),
|
350 |
+
gr.Slider(1, 10, value=5, step=1, label="Stroke Width"),
|
351 |
],
|
352 |
outputs=[
|
353 |
gr.Image(label="Roboflow Output Image"),
|