Spaces:

chlab
/

interactive_kinematic_planet_detector

Sleeping

jpterry commited on Jun 17, 2022

Commit

8f4303a

1 Parent(s): 3f9e954

bug fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -117,8 +117,9 @@ def get_activations(model, image: list, model_name: str,
     layer_outputs = {}
     for i in range(len(model.model.features)):
-        image = model.model.features[i](image)
         if i in activation_indices[model_name]:
             layer_outputs[i] = image
             print(i, layer_outputs[i].shape)

     layer_outputs = {}
+    temp_image = image
     for i in range(len(model.model.features)):
+        temp_image = model.model.features[i](temp_image)
         if i in activation_indices[model_name]:
             layer_outputs[i] = image
             print(i, layer_outputs[i].shape)