app, model, examples

.gitattributes

@@ -32,3 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,35 @@
+import gradio as gr
+import yolov5
+from PIL import Image
+
+# Model
+model = yolov5.load('./yolo_datacat/best.pt')
+
+def predict(im, threshold=0.50):
+    # g = (size / max(im.size))  # gain
+    # im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)  # resize
+    model.conf = threshold
+    results = model(im)  # inference
+    numpy_image = results.render()[0]
+    output_image = Image.fromarray(numpy_image)
+    return output_image
+
+title = "YOLOv5 - Auction sale catalogues layout analysis"
+description = "<p style='text-align: center'>YOLOv5 Gradio demo for auction sales catalogues layout analysis. Detecting titles and catalogues entries.</p>"
+article = "<p style='text-align: center'>YOLOv5 source code : <a href='https://github.com/ultralytics/yolov5'>Source code</a> | <a href='https://pytorch.org/hub/ultralytics_yolov5'>PyTorch Hub</a></p>"
+examples = [['./img_examples/12148-bpt6k1240127r.pdf_page_20.png', 0.50],
+ ['./img_examples/12148-bpt6k1240127r.pdf_page_21.png', 0.50],
+  ['./img_examples/12148-bpt6k1240127r.pdf_page_27.png', 0.50]]
+
+demo=gr.Interface(fn=predict,
+            inputs=[gr.Image(type="pil", label="document image"), gr.Slider(maximum=1, step=0.01, value=0.50)],
+            outputs=gr.Image(type="pil", label="annotated document").style(height=700),
+            title=title,
+            description=description,
+            article=article,
+            examples=examples,
+            theme="huggingface")
+
+
+if __name__ == "__main__":
+    demo.launch(debug=True)

requirements.txt

@@ -0,0 +1,2 @@
+yolov5==7.0.2
+gradio

yolo_datacat/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0a2d2257828cc28b228a06b63f88daf0a2c42f725735d551a66ae5cd7174e75e
+size 173122433