pil image as input

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from PIL import Image,ImageDraw, ImageFont, ImageOps
 import sys
 import torch
-from util import Detection
+from util import Detection, load_font
 import os
 
 if os.environ.get('FACE_MODEL') is not None:
@@ -17,16 +17,17 @@ sys.path.append("./yolov5")
 
 from yolov5.detect import predict, load_yolo_model
 
+
 # Load Models
 model, stride, names, pt, jit, onnx, engine = load_yolo_model("face_model.pt", imgsz=[320,320])
 age_model_ts = torch.jit.load("age_model.pt")
 
-roboto_font = ImageFont.truetype("Roboto-Regular.ttf")
+roboto_font = load_font(height_px=18)
 
-def run_yolo(img):
+def run_yolo(img0):
 
-    img_path = img
-    img0 = Image.open(img_path).convert("RGB")
+    #img_path = img
+    #img0 = Image.open(img_path).convert("RGB")
 
     img0 = ImageOps.contain(img0, (720,720))
     img0 = ImageOps.exif_transpose(img0)
@@ -55,15 +56,16 @@ def run_yolo(img):
 
         detections.append(det)
         draw.rectangle(((det.xmin, det.ymin), (det.xmax, det.ymax)), fill=None, outline=(255,255,255))
-        draw.rectangle(((det.xmin, det.ymin - 10), (det.xmax, det.ymin)), fill=(255,255,255))
-        draw.text((det.xmin, det.ymin - 10), det.class_name, fill=(0,0,0), font=roboto_font)
+        draw.rectangle(((det.xmin, det.ymin - 20), (det.xmax, det.ymin)), fill=(255,255,255))
+        draw.text((det.xmin, det.ymin - 20), det.class_name, fill=(0,0,0), font=roboto_font)
 
+    # img0.save("img.jpg")
     return img0
 
 
-# run_yolo("D:\\Download\\IMG_20220803_153335.jpg")
-
-inputs = gr.inputs.Image(type='filepath', label="Input Image")
+#run_yolo("D:\\Download\\IMG_20220803_153335c.jpg")
+#sys.exit(1)
+inputs = gr.inputs.Image(type='pil', label="Input Image")
 outputs = gr.outputs.Image(type="pil", label="Output Image")
 
 title = "AgeGuesser"

util.py

@@ -1,5 +1,7 @@
-class Detection(object):
+from PIL import Image,ImageDraw, ImageFont, ImageOps
+
 
+class Detection(object):
 
     def __init__(self, id: int, xmin: int, ymin: int, xmax:int, ymax:int, conf: float, class_id:int, class_name:str, orig_img_sz: "tuple[int]") -> None:
         
@@ -79,4 +81,17 @@ class Detection(object):
         return iou
 
     def __str__(self) -> str:
-        return f"[{self.xmin}, {self.ymin}, {self.xmax}, {self.ymax}]"
+        return f"[{self.xmin}, {self.ymin}, {self.xmax}, {self.ymax}]"
+
+
+def load_font(height_px = 20):
+
+    init_size = 12
+    roboto_font = ImageFont.truetype("Roboto-Regular.ttf", size=init_size)
+
+    while roboto_font.getsize("20")[1] < height_px:
+
+        init_size += 1
+        roboto_font = ImageFont.truetype("Roboto-Regular.ttf", size=init_size)
+    
+    return roboto_font