jpterry commited on
Commit
56c8acc
·
1 Parent(s): 86d1c7a

minor updates

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -166,6 +166,8 @@ def predict_and_analyze(model_name, num_channels, dim, image):
166
 
167
  # plot raw activations
168
  output, input_image, activation_1, activation_2 = get_activations(model, image, sub_mean=False)
 
 
169
  fig2 = plot_activations(activation_1, activation_2, origin=origin)
170
 
171
 
@@ -173,7 +175,7 @@ def predict_and_analyze(model_name, num_channels, dim, image):
173
  plt.rcParams['xtick.labelsize'] = ticks
174
  plt.rcParams['ytick.labelsize'] = ticks
175
 
176
- input_fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(15, 8))
177
 
178
  im0 = ax.imshow(input_image, cmap=cmap,
179
  origin=origin)
 
166
 
167
  # plot raw activations
168
  output, input_image, activation_1, activation_2 = get_activations(model, image, sub_mean=False)
169
+ activation_1 = normalize_array(activation_1)
170
+ activation_2 = normalize_array(activation_2)
171
  fig2 = plot_activations(activation_1, activation_2, origin=origin)
172
 
173
 
 
175
  plt.rcParams['xtick.labelsize'] = ticks
176
  plt.rcParams['ytick.labelsize'] = ticks
177
 
178
+ input_fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(19, 8))
179
 
180
  im0 = ax.imshow(input_image, cmap=cmap,
181
  origin=origin)