added files

app.py

@@ -0,0 +1,53 @@
+'''
+File: recognize.py
+Project: PatternsRecognizer
+Author: Milko Videv (milko.videv@thalesgroup.com)
+-----
+Last Modified: Monday, 4th March 2024 9:38:39 am
+Modified By: Milko Videv (milko.videv@thalesgroup.com>)
+-----
+Copyright 2017 - 2024, Thales DIS, MCS SSH
+-----
+HISTORY:
+Date      	By	Comments
+----------	---	---------------------------------------------------------
+'''
+
+import gradio as gr
+from predict import *
+from tools import *
+from gradio.themes.utils.colors import slate  # type: ignore
+
+def render():
+    
+    title = "Recognizer of Bulgarian, Indian and Japanese design patterns"
+    description = "Select image or drag some of the examples below."    
+    examples = [
+        './samples/bulgarian.jpg', 
+        './samples/indian.jpg', 
+        './samples/japanese.jpg'
+    ]              
+    rand_examples = [
+        pick_random_file("./patterns/bulgarian"),
+        pick_random_file("./patterns/indian"),
+        pick_random_file("./patterns/japanese"),
+    ] 
+      
+    demo = gr.Interface(fn=predict,
+        theme=gr.themes.Monochrome(primary_hue=slate),
+        inputs=gr.components.Image(shape=(512, 512), interactive=True),
+        outputs=gr.components.Label(num_top_classes=3),
+        title=title,
+        description=description,
+        examples=examples,
+        allow_flagging="manual",
+        flagging_options=["Correct", "Incorrect"],
+        css=
+        "#component-3 { background-color: rgb(192, 192, 192) !important; }"
+        "#component-3 H1 { margin: 1.5rem 0 1.5rem 0; color: #252873; !important;}"
+        "footer { display: none !important; }"         
+        )
+    demo.queue().launch()
+
+if __name__ == "__main__":
+    render()

models/trained.pkl

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

predict.py

@@ -0,0 +1,45 @@
+'''
+File: recognize.py
+Project: PatternsRecognizer
+Author: Milko Videv (milko.videv@thalesgroup.com)
+-----
+Last Modified: Tuesday, 5th March 2024 10:02:08 am
+Modified By: Milko Videv (milko.videv@thalesgroup.com>)
+-----
+Copyright 2017 - 2024, Thales DIS, MCS SSH
+-----
+HISTORY:
+Date      	By	Comments
+----------	---	---------------------------------------------------------
+'''
+
+from fastcore.all import *
+from fastai.vision.all import *
+from tools import *
+
+learn = load_learner("./models/trained.pkl")
+labels = learn.dls.vocab
+
+def predict(image):    
+    img = PILImage.create(image)
+    pred,pred_idx,probs = learn.predict(img)
+    result = {labels[i]: float(probs[i]) for i in range(len(labels))}
+
+    max_type = max(result, key=result.get)
+    max_probability = result[max_type] * 100
+    print(f"{max_type} with probability {max_probability:.2f}%")
+
+    return result
+
+if __name__ == "__main__":
+    if debugger_is_active():
+        predict("./POC/PatternsRecognizer/samples/bulgarian.jpg")
+        sys.exit()
+    else:
+        if len(sys.argv) < 2:
+            print("Use: python predict.py <image path>")   
+            print("Example: python predict.py samples/bulgarian.jpg")
+            sys.exit()
+
+        src = sys.argv[1]    
+        predict(src)

tools.py

@@ -0,0 +1,62 @@
+'''
+File: tools.py
+Project: PatternsRecognizer
+Author: Milko Videv (milko.videv@thalesgroup.com)
+-----
+Last Modified: Saturday, 2nd March 2024 10:25:48 am
+Modified By: Milko Videv (milko.videv@thalesgroup.com>)
+-----
+Copyright 2017 - 2024, Thales DIS, MCS SSH
+-----
+HISTORY:
+Date      	By	Comments
+----------	---	---------------------------------------------------------
+'''
+import sys 
+import time
+import os
+import random
+
+def get_yes_no(prompt):
+    while True:
+        response = input(prompt + " [y]|n: ").strip().lower()
+        if response == "" or response in ["yes", "y"]:
+            return True
+        elif response in ["no", "n"]:
+            return False
+        else:
+            print("\nPlease enter 'yes|y|no|n': ")
+
+def debugger_is_active() -> bool:
+    return hasattr(sys, 'gettrace') and sys.gettrace() is not None
+
+def pick_random_file(directory):
+    files = [file for file in os.listdir(directory) if os.path.isfile(os.path.join(directory, file))]
+    if files:
+        return os.path.join(directory, random.choice(files))
+    else:
+        return None
+    
+class Stopwatch:
+    def __init__(self):
+        self.start_time = None
+        self.end_time = None
+
+    def start(self):
+        self.start_time = time.time()
+
+    def stop(self):
+        self.end_time = time.time()
+
+    def elapsed_time(self):
+        if self.start_time is None:
+            return 0
+        if self.end_time is None:
+            elapsed = time.time() - self.start_time
+        else:
+            elapsed = self.end_time - self.start_time
+        return elapsed
+
+    def reset(self):
+        self.start_time = None
+        self.end_time = None