bug fix

app.py

@@ -83,7 +83,6 @@ def get_activations(intermediate_model, image: list,
     in_image = np.sum(image[0, :, :, :], axis=0)
     in_image = normalize_array(in_image)
     
-    
     # im1 = ax1.imshow(in_image, cmap=cmap, vmin=0, vmax=vmax, origin=origin)
     if layer is None:
         activation_1 = np.sum(output_1[0, :, :, :], axis=0)
@@ -115,7 +114,7 @@ def get_activations(intermediate_model, image: list,
     
     # plt.show()
     
-    return in_image, activation_1, activation_2
+    return output, in_image, activation_1, activation_2
 
 
 def predict_and_analyze(model_name, num_channels, dim, image):
@@ -163,8 +162,6 @@ def predict_and_analyze(model_name, num_channels, dim, image):
         image = image[np.newaxis, :, :, :]
         
     assert image.shape == (1, num_channels, W, W), "Data is the wrong shape"
-        
-    input_image = np.sum(image[0, :, :, :], axis=0)
     
     model_name += '_%i' % (num_channels)
         
@@ -173,7 +170,7 @@ def predict_and_analyze(model_name, num_channels, dim, image):
     model = load_model(model_name, activation=True)
     
     print("Looking at activations")
-    output, activation_1, activation_2 = get_activations(model, image, sub_mean=True)
+    output, input_image, activation_1, activation_2 = get_activations(model, image, sub_mean=True)
     
     print(output)