bug fix
Browse files
app.py
CHANGED
@@ -83,7 +83,7 @@ def get_activations(model, image: list, model_name: str,
|
|
83 |
|
84 |
|
85 |
layer_outputs = {}
|
86 |
-
for i in range(len(model.
|
87 |
image = model.features[i](image)
|
88 |
layer_outputs[i] = image
|
89 |
print(i, layer_outputs[i].shape)
|
|
|
83 |
|
84 |
|
85 |
layer_outputs = {}
|
86 |
+
for i in range(len(model.features)):
|
87 |
image = model.features[i](image)
|
88 |
layer_outputs[i] = image
|
89 |
print(i, layer_outputs[i].shape)
|