x

app.py

@@ -25,8 +25,8 @@ descriptions = json.load(open('valid_list.json', 'r'))
 # example_idx = int(col1.number_input('Sample an example (description + corresponding images) from the validation set', value=0, min_value=0, max_value=len(descriptions)-1))
 if col1.button('Sample an example (description + corresponding images) from the validation set'):
     st.session_state.example_idx = random.randint(0, len(descriptions)-1)
-img_set, idx, descr = descriptions[st.session_state.example_idx]
-idx = int(idx)
+img_set, true_idx, descr = descriptions[st.session_state.example_idx]
+true_idx = int(true_idx)
 images = [prefix+'/'+img_set+'/'+i for i in set2ids[img_set]]
 img_urls = images.copy()
 index = int(col2.number_input('Image Index from 0 to 9', value=0, min_value=0, max_value=9))
@@ -37,21 +37,27 @@ if col1.button('Click to reveal/hide groundtruth image index (try to guess yours
 col1.markdown(f'**Description**:')
 col1.markdown(f'**{descr}**') 
 
-img = images[index]
-images[index] = ImageOps.expand(Image.open(io.BytesIO(requests.get(images[index], stream=True).content)),border=20,fill='blue')
-
+big_img = images[index]
+img = Image.open(io.BytesIO(requests.get(images[index], stream=True).content))
+img_width, img_height = img.size
+smaller = min(img_width, img_height)
+images[index]= ImageOps.expand(img,border=smaller//18,fill='blue')
 
 caps = list(range(10))
 cap = str(index)
 
 if st.session_state.show:
-    caps[idx] = f'{idx} (TARGET IMAGE)'
-    if idx == index:
-        cap = f'{idx} (TARGET IMAGE)'
+    caps[true_idx] = f'{true_idx} (TARGET IMAGE)'
+    img = Image.open(io.BytesIO(requests.get(img_urls[index], stream=True).content))
+    img_width, img_height = img.size
+    smaller = min(img_width, img_height)
+    images[true_idx] = ImageOps.expand(img,border=smaller//12,fill='green')
+    if true_idx == index:
+        cap = f'{true_idx} (TARGET IMAGE)'
 else:
-    caps[idx] = f'{idx}'
-    if idx == index:
-        cap = f'{idx}'
+    caps[true_idx] = f'{true_idx}'
+    if true_idx == index:
+        cap = f'{true_idx}'
 
-col1.image(img, use_column_width=True, caption=cap)
+col1.image(big_img, use_column_width=True, caption=cap)
 col2.image(images, width=175, caption=caps)