save image id

app.py

@@ -29,6 +29,7 @@ def main():
         title = gr.Markdown("# Saliency evaluation - experiment 1")
         user_state = gr.State(0)
         answers = gr.State([])
+        img_ids = gr.State([])
         start_time = gr.State(time.time())
 
         concepts = load_csv_concepts(data_dir)
@@ -68,11 +69,12 @@ def main():
         with gr.Row():
             count = user_state if isinstance(user_state, int) else user_state.value
             images = load_image_and_saliency(count, data_dir)
-            target_img = gr.Image(images[0], elem_classes="main-image delay", visible=False)
-            saliency_gradcam = gr.Image(images[1], elem_classes="main-image", visible=False)
-            saliency_lime = gr.Image(images[2], elem_classes="main-image", visible=False)
-            saliency_sidu = gr.Image(images[4], elem_classes="main-image", visible=False)
-            saliency_rise = gr.Image(images[3], elem_classes="main-image", visible=False)
+            img_ids = gr.State([images[0]])  
+            target_img = gr.Image(images[1], elem_classes="main-image delay", visible=False)
+            saliency_gradcam = gr.Image(images[2], elem_classes="main-image", visible=False)
+            saliency_lime = gr.Image(images[3], elem_classes="main-image", visible=False)
+            saliency_sidu = gr.Image(images[5], elem_classes="main-image", visible=False)
+            saliency_rise = gr.Image(images[4], elem_classes="main-image", visible=False)
             
                 
         with gr.Row():
@@ -88,7 +90,7 @@ def main():
         def update_images(user_state):
             count = user_state if isinstance(user_state, int) else user_state.value
             if count < config['dataset'][config['dataset']['name']]['n_classes']:
-                images = load_image_and_saliency(count, data_dir)
+                #images = load_image_and_saliency(count, data_dir)
                 
                 # image examples
                 images = load_example_images(count, data_dir)
@@ -112,18 +114,19 @@ def main():
             else:
                 return img1, img2, img3, img4, img5, img6, img7, img8, img9, img10, img11, img12, img13, img14, img15, img16
 
-        def update_saliencies(dropdown1, dropdown2, dropdown3, dropdown4, user_state):
+        def update_saliencies(dropdown1, dropdown2, dropdown3, dropdown4, user_state, img_ids):
             count = user_state if isinstance(user_state, int) else user_state.value
             if count < config['dataset'][config['dataset']['name']]['n_classes']:
                 images = load_image_and_saliency(count, data_dir)
+                img_ids.append(images[0])  
                 target_img = gr.Image(images[0], elem_classes="main-image", visible=True)
                 saliency_gradcam = gr.Image(images[1], elem_classes="main-image", visible=True)
                 saliency_lime = gr.Image(images[2], elem_classes="main-image", visible=True)
                 saliency_sidu = gr.Image(images[4], elem_classes="main-image", visible=True)
                 saliency_rise = gr.Image(images[3], elem_classes="main-image", visible=True)
-                return target_img, saliency_gradcam, saliency_lime, saliency_rise, saliency_sidu
+                return img_ids, target_img, saliency_gradcam, saliency_lime, saliency_rise, saliency_sidu
             else:
-                return target_img, saliency_gradcam, saliency_lime, saliency_rise, saliency_sidu
+                return img_ids, target_img, saliency_gradcam, saliency_lime, saliency_rise, saliency_sidu
 
         def update_state(state):
             count = state if isinstance(state, int) else state.value
@@ -264,8 +267,8 @@ def main():
             outputs=target_img_label
         ).then(
             update_saliencies, 
-            inputs=[dropdown1, dropdown2, dropdown3, dropdown4, user_state], 
-            outputs={target_img, saliency_gradcam, saliency_lime, saliency_sidu, saliency_rise},
+            inputs=[dropdown1, dropdown2, dropdown3, dropdown4, user_state, img_ids],
+            outputs={img_ids, target_img, saliency_gradcam, saliency_lime, saliency_sidu, saliency_rise},
         ).then(
             update_questions,
             inputs=user_state,

src/utils.py

@@ -19,7 +19,7 @@ def load_image_and_saliency(class_idx, data_dir):
     lime_image = os.path.join(data_dir, 'saliency', 'lime', images[id])
     sidu_image = os.path.join(data_dir, 'saliency', 'sidu', images[id])
     rise_image = os.path.join(data_dir, 'saliency', 'rise', images[id])
-    return image, gradcam_image, lime_image, sidu_image, rise_image
+    return id, image, gradcam_image, lime_image, sidu_image, rise_image
 
 def load_example_images(class_idx, data_dir, max_images=16):
     path = os.path.join(data_dir, 'images', str(class_idx))