[pycaffe] no need to squeeze output after #1970

fix #2041 reported by @dgmp88
BVLC · Mar 8, 2015 · fc35930 · fc35930
1 parent 256366b
commit fc35930
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/caffe/classifier.py b/python/caffe/classifier.py
@@ -83,7 +83,7 @@ def predict(self, inputs, oversample=True):
         for ix, in_ in enumerate(input_):
             caffe_in[ix] = self.transformer.preprocess(self.inputs[0], in_)
         out = self.forward_all(**{self.inputs[0]: caffe_in})
-        predictions = out[self.outputs[0]].squeeze(axis=(2,3))
+        predictions = out[self.outputs[0]]
 
         # For oversampling, average predictions across crops.
         if oversample: