Update to use ONNX

app.py

@@ -49,14 +49,16 @@ class Model:
         return {name: outputs[i] for i, name in enumerate(self.output_names)}
 
 def draw_boxes(image: Image.Image, outputs: dict):
-    draw = ImageDraw.Draw(image)
+    draw = ImageDraw.Draw(image, "RGBA")  # Use RGBA for transparency
 
-    # Dynamic font size based on image width
+    # Dynamic font size based on image dimensions
     image_width, image_height = image.size
-    font_size = max(15, image_width // 100)  # Adjust as needed
+    font_size = max(20, image_width // 50)  # Increased minimum font size
     try:
+        # Attempt to load a truetype font; adjust the path if necessary
         font = ImageFont.truetype("arial.ttf", size=font_size)
     except IOError:
+        # Fallback to default font if truetype font is not found
         font = ImageFont.load_default()
 
     boxes = outputs.get('detected_boxes', [])
@@ -76,7 +78,7 @@ def draw_boxes(image: Image.Image, outputs: dict):
         bottom = ymax * image_height
 
         # Draw bounding box
-        draw.rectangle([left, top, right, bottom], outline="red", width=2)
+        draw.rectangle([left, top, right, bottom], outline="red", width=3)
 
         # Prepare label text
         text = f"{label}: {score:.2f}"
@@ -86,11 +88,24 @@ def draw_boxes(image: Image.Image, outputs: dict):
         text_width = text_bbox[2] - text_bbox[0]
         text_height = text_bbox[3] - text_bbox[1]
 
-        # Draw rectangle behind text for better visibility
-        draw.rectangle([left, top - text_height - 4, left + text_width + 4, top], fill="red")
+        # Calculate label background position
+        # Ensure the label box does not go above the image
+        label_top = max(top - text_height - 10, 0)
+        label_left = left
+
+        # Draw semi-transparent rectangle behind text
+        draw.rectangle(
+            [label_left, label_top, label_left + text_width + 10, label_top + text_height + 10],
+            fill=(255, 0, 0, 160)  # Semi-transparent red
+        )
 
         # Draw text
-        draw.text((left + 2, top - text_height - 2), text, fill="white", font=font)
+        draw.text(
+            (label_left + 5, label_top + 5),
+            text,
+            fill="white",
+            font=font
+        )
 
     return image
 
@@ -121,11 +136,16 @@ def detect_objects(image):
 iface = gr.Interface(
     fn=detect_objects,
     inputs=gr.Image(type="pil"),
-    outputs=[gr.Image(type="pil", label="Detected Objects"), gr.Textbox(label="Detections")],
+    outputs=[
+        gr.Image(type="pil", label="Detected Objects"),
+        gr.Textbox(label="Detections")
+    ],
     title="Object Detection with ONNX Model",
     description="Upload an image to detect objects using the ONNX model.",
     examples=["examples/card1.jpg", "examples/card2.jpg", "examples/card3.jpg"],
-    theme="default"  # You can choose other themes if desired
+    theme="default",  # You can choose other themes if desired
+    allow_flagging="never",  # Disable flagging if not needed
+    layout="horizontal"  # Display outputs side by side
 )
 
 if __name__ == "__main__":