From ebbaebc608d923f437f396957d71615493c13055 Mon Sep 17 00:00:00 2001
From: Matt Harvey <harveym@gmail.com>
Date: Sun, 16 Apr 2017 13:23:14 -0700
Subject: [PATCH] Change feature return
---
extractor.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/extractor.py b/extractor.py
index d9ca868..a3d8cd8 100644
--- 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]
--
GitLab