Change feature return

ebbaebc6 · Matt Harvey · c516a622 · ebbaebc6
Commit ebbaebc6 authored 7 years ago by Matt Harvey
Hide whitespace changes
Inline Side-by-side

Showing

with 1 addition and 1 deletion
+1 -1
--- a/extractor.py
+++ b/extractor.py
@@ -47,7 +47,7 @@ class Extractor():

        if self.weights is None:
            # For imagenet/default network:
-            features = features[0][0][0]
+            features = features[0]
        else:
            # For loaded network:
            features = features[0]