[pycaffe] classifier + detector only have one input

anshumang · Mar 8, 2015 · 0b39892 · 0b39892
1 parent 3f68c51
commit 0b39892
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/caffe/classifier.py b/python/caffe/classifier.py
@@ -28,7 +28,7 @@ def __init__(self, model_file, pretrained_file, image_dims=None,
         # configure pre-processing
         in_ = self.inputs[0]
         self.transformer = caffe.io.Transformer(
-            {in_: self.blobs[in_].data.shape for in_ in self.inputs})
+            {in_: self.blobs[in_].data.shape})
         self.transformer.set_transpose(in_, (2,0,1))
         if mean is not None:
             self.transformer.set_mean(in_, mean)

diff --git a/python/caffe/detector.py b/python/caffe/detector.py
@@ -40,7 +40,7 @@ def __init__(self, model_file, pretrained_file, mean=None,
         # configure pre-processing
         in_ = self.inputs[0]
         self.transformer = caffe.io.Transformer(
-            {in_: self.blobs[in_].data.shape for in_ in self.inputs})
+            {in_: self.blobs[in_].data.shape})
         self.transformer.set_transpose(in_, (2,0,1))
         if mean is not None:
             self.transformer.set_mean(in_, mean)