add images 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
+*.jpg filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,2 @@
+venv/
+flagged/

.idea/.gitignore

@@ -0,0 +1,8 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+# Editor-based HTTP Client requests
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml

.idea/YOLOv5_medieval_register.iml

@@ -0,0 +1,14 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$">
+      <excludeFolder url="file://$MODULE_DIR$/venv" />
+    </content>
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+  <component name="PyDocumentationSettings">
+    <option name="format" value="PLAIN" />
+    <option name="myDocStringFormat" value="Plain" />
+  </component>
+</module>

.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,12 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="DuplicatedCode" enabled="true" level="WEAK WARNING" enabled_by_default="true">
+      <Languages>
+        <language minSize="85" name="Python" />
+      </Languages>
+    </inspection_tool>
+    <inspection_tool class="Eslint" enabled="true" level="WARNING" enabled_by_default="true" />
+    <inspection_tool class="JSHint" enabled="true" level="ERROR" enabled_by_default="true" />
+  </profile>
+</component>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml

@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.11 (YOLOv5_medieval_register)" project-jdk-type="Python SDK" />
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/YOLOv5_medieval_register.iml" filepath="$PROJECT_DIR$/.idea/YOLOv5_medieval_register.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>

app.py

@@ -1,17 +1,17 @@
 """
-Gradio APP for Yolo segmentation model of medival registers.
+Gradio APP for Yolov5 medival registers layout analysis.
 
 Date: 20/12/2022
 Author: lterriel
 
-Test segmentation model with object-detection Yolo model
-App inspired by HugoSchtr/DataCat_Yolov5
+* Gradio app adapted from HugoSchtr/DataCat_Yolov5
 """
+import glob
 
 import gradio as gr
 import yolov5
 from PIL import Image
-import torch
+# import torch
 from huggingface_hub import hf_hub_download
 
 # Add new models here
@@ -33,22 +33,21 @@ def yolo_inference(im, model_path, threshold=0.50):
     return output_image
 
 title = "YOLOv5 - Medieval Register Segmentation"
-
-#examples = [['', 0.50],
-# ['', 0.50],
-#  ['', 0.50]]
+description = "<h5 style='text-align: center;'>YOLOv5 Gradio demo for medieval register layout analysis.</h3><br><ul><li>YOLOv5 source code : <a href='https://github.com/ultralytics/yolov5'>Source code</a></li><li>The examples provided are digitized images extracted from the <i>registers of the chapter of Notre-Dame (1326-1504)</i> available on the <a href='https://www.siv.archives-nationales.culture.gouv.fr/siv/rechercheconsultation/consultation/ir/consultationIR.action?irId=FRAN_IR_059635' target='_blank'>SIV of the National Archives of France</li></ul>"
 
 inputs = [gr.Image(type="pil", label="document image"),
-          gr.inputs.Dropdown(model_names, label="Model"),
+          gr.inputs.Dropdown(model_names, label="Model", default=model_names[0]),
           gr.Slider(maximum=1, step=0.01, value=0.50)]
 
+examples = [[file, "0.50"] for file in glob.glob("./images_examples/*.jpg")]
+
 demo=gr.Interface(fn=yolo_inference,
                   inputs=inputs,
                   outputs=gr.Image(type="pil", label="annotated document").style(height=800),
                   title=title,
-                  #examples=examples,
-                  theme="huggingface")
-
+                  description=description,
+                  theme="huggingface",
+                  examples=examples)
 
 if __name__ == "__main__":
     demo.launch(debug=True)