Spaces:

chlab
/

interactive_kinematic_planet_detector

Sleeping

jpterry commited on Jun 17, 2022

Commit

04b6f1f

1 Parent(s): 25fdf2a

bug fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -121,7 +121,7 @@ def get_activations(model, image: list, model_name: str,
     for i in range(len(model.features)):
         temp_image = model.features[i](temp_image)
         if i in activation_indices[model_name]:
-            layer_outputs[i] = image
             # print(i, layer_outputs[i].shape)
         if i == max(activation_indices[model_name]):
             break

     for i in range(len(model.features)):
         temp_image = model.features[i](temp_image)
         if i in activation_indices[model_name]:
+            layer_outputs[i] = temp_image
             # print(i, layer_outputs[i].shape)
         if i == max(activation_indices[model_name]):
             break