Update app.py

app.py

@@ -70,13 +70,10 @@ def load_or_augment_images(img1_input, img2_input, use_aug):
 
 def run_occlusion(w_size, stride):
   heatmap1, heatmap2 = occlusion(img1, img2, ssl_model, w_size = 64, stride = 8, batch_size = 32)
-  heatmap1_ca, heatmap2_ca = occlusion_context_agnositc(img1, img2, ssl_model, w_size = 64, stride = 8, batch_size = 32)
   heatmap1_po, heatmap2_po = pairwise_occlusion(img1, img2, ssl_model, batch_size = 32, erase_scale = (0.1, 0.3), erase_ratio = (1, 1.5), num_erases = 100)
 
   added_image1 = overlay_heatmap(img1, heatmap1, denormalize = denorm)
   added_image2 = overlay_heatmap(img2, heatmap2, denormalize = denorm)
-  added_image1_ca = overlay_heatmap(img1, heatmap1_ca, denormalize = denorm)
-  added_image2_ca = overlay_heatmap(img2, heatmap2_ca, denormalize = denorm)
 
   fig, axs = plt.subplots(2, 4, figsize=(20,10))
   np.vectorize(lambda ax:ax.axis('off'))(axs)
@@ -84,14 +81,11 @@ def run_occlusion(w_size, stride):
   axs[0, 0].imshow(show_image(img1, denormalize = denorm))
   axs[0, 1].imshow(added_image1)
   axs[0, 1].set_title("Conditional Occlusion")
-  axs[0, 2].imshow(added_image1_ca)
-  axs[0, 2].set_title("CA Cond. Occlusion")
-  axs[0, 3].imshow((deprocess(img1, denormalize = denorm) * heatmap1_po[:,:,None]).astype('uint8'))
-  axs[0, 3].set_title("Pairwise Occlusion")
+  axs[0, 2].imshow((deprocess(img1, denormalize = denorm) * heatmap1_po[:,:,None]).astype('uint8'))
+  axs[0, 2].set_title("Pairwise Occlusion")
   axs[1, 0].imshow(show_image(img2, denormalize = denorm))
   axs[1, 1].imshow(added_image2)
-  axs[1, 2].imshow(added_image2_ca)
-  axs[1, 3].imshow((deprocess(img2, denormalize = denorm) * heatmap2_po[:,:,None]).astype('uint8'))
+  axs[1, 2].imshow((deprocess(img2, denormalize = denorm) * heatmap2_po[:,:,None]).astype('uint8'))
   plt.subplots_adjust(wspace=0, hspace = 0.01)
   pil_output = fig2img(fig)
   return pil_output