Arranged Markdown

app.py

@@ -11,7 +11,25 @@ ALPHA = 0.3
 GENERATORS = ['itti', 'deepgaze']
 
 MARKDOWN = """
-<h1 style='text-align: center'>Saliency Ranking: Itti vs. Deepgaze</h1>
+<h1 style='text-align: center'>Saliency Ranking 🥇</h1>
+
+Saliency Ranking is a fundamental 🌟 **Computer Vision** 🌟 process aimed at discerning the most visually significant features within an image 🖼️.
+
+🌟 This demo showcases the **SaRa (Saliency-Driven Object Ranking)** model for Saliency Ranking 🎯, which can efficiently rank the visual saliency of an image without requiring any training. 🖼️
+
+This technique is based on the Saliency Map generator model from Itti, which works on the primate visual cortex 🧠, and can work with or without depth information 🔄.
+
+<div style="display: flex; align-items: center;">
+    <a href="https://github.com/dylanseychell/SaliencyRanking" style="margin-right: 10px;">
+      <img src="https://badges.aleen42.com/src/github.svg">
+    </a>
+    <a href="https://github.com/mbar0075/SaRa" style="margin-right: 10px;">
+      <img src="https://badges.aleen42.com/src/github.svg">
+    </a>
+    <a href="https://github.com/matthewkenely/ICT3909" style="margin-right: 10px;">
+      <img src="https://badges.aleen42.com/src/github.svg">
+    </a>
+  </div>
 """
 
 IMAGE_EXAMPLES = [
@@ -23,7 +41,8 @@ IMAGE_EXAMPLES = [
 def detect_and_annotate(image, 
                         GRID_SIZE, 
                         generator, 
-                        ALPHA=ALPHA)-> np.ndarray:
+                        ALPHA=ALPHA,
+                        mode=1)-> np.ndarray:
     # Converting from PIL to OpenCV
     image = np.array(image)
     # Convert image from BGR to RGB
@@ -37,7 +56,7 @@ def detect_and_annotate(image,
     sara.reset()
 
     # Running sara (Original implementation on itti)
-    sara_info = sara.return_sara(sara_image, GRID_SIZE, generator, mode=1)
+    sara_info = sara.return_sara(sara_image, GRID_SIZE, generator, mode=mode)
 
     # Generate saliency map
     saliency_map = sara.return_saliency(image, generator=generator)
@@ -67,14 +86,17 @@ def process_image(
 
     itti_saliency_map, itti_heatmap = detect_and_annotate(
         input_image, GRIDSIZE, 'itti')
-    deepgaze_saliency_map, deepgaze_heatmap = detect_and_annotate(
-        input_image, GRIDSIZE, 'deepgaze')
+    _, itti_heatmap2 = detect_and_annotate(
+        input_image, GRIDSIZE, 'itti', mode=2)
+    # deepgaze_saliency_map, deepgaze_heatmap = detect_and_annotate(
+    #     input_image, GRIDSIZE, 'deepgaze')
 
     return (
         itti_saliency_map,
         itti_heatmap,
-        deepgaze_saliency_map,
-        deepgaze_heatmap,
+        itti_heatmap2,
+        # deepgaze_saliency_map,
+        # deepgaze_heatmap,
     )
 
 grid_size_Component = gr.Slider(
@@ -101,24 +123,28 @@ with gr.Blocks() as demo:
             type='pil',
             label='Input'
         )
-    with gr.Row():
         itti_saliency_map = gr.Image(
             type='pil',
             label='Itti Saliency Map'
         )
+    with gr.Row():
         itti_heatmap = gr.Image(
             type='pil',
             label='Itti Saliency Ranking Heatmap'
         )
-    with gr.Row():
-        deepgaze_saliency_map = gr.Image(
+        itti_heatmap2 = gr.Image(
             type='pil',
-            label='DeepGaze Saliency Map'
-        )
-        deepgaze_heatmap = gr.Image(
-            type='pil',
-            label='DeepGaze Saliency Ranking Heatmap'
+            label='Itti Saliency Ranking Heatmap'
         )
+    # with gr.Row():
+    #     deepgaze_saliency_map = gr.Image(
+    #         type='pil',
+    #         label='DeepGaze Saliency Map'
+    #     )
+    #     deepgaze_heatmap = gr.Image(
+    #         type='pil',
+    #         label='DeepGaze Saliency Ranking Heatmap'
+    #     )
     submit_button_component = gr.Button(
         value='Submit',
         scale=1,
@@ -134,8 +160,9 @@ with gr.Blocks() as demo:
         outputs=[
             itti_saliency_map,
             itti_heatmap,
-            deepgaze_saliency_map,
-            deepgaze_heatmap,
+            itti_heatmap2,
+            # deepgaze_saliency_map,
+            # deepgaze_heatmap,
         ]
     )
 
@@ -148,8 +175,9 @@ with gr.Blocks() as demo:
         outputs=[
             itti_saliency_map,
             itti_heatmap,
-            deepgaze_saliency_map,
-            deepgaze_heatmap,
+            itti_heatmap2,
+            # deepgaze_saliency_map,
+            # deepgaze_heatmap,
         ]
     )