diff --git a/python/caffe/classifier.py b/python/caffe/classifier.py index ea29fed86f9..983760a786d 100644 --- a/python/caffe/classifier.py +++ b/python/caffe/classifier.py @@ -92,7 +92,7 @@ def predict(self, inputs, oversample=True): # For oversampling, average predictions across crops. if oversample: - predictions = predictions.reshape((len(predictions) / 10, 10, -1)) + predictions = predictions.reshape((len(predictions) // 10, 10, -1)) predictions = predictions.mean(1) return predictions