modified division operator for compatibility of python 3

jihochu · Sep 6, 2017 · 3dad332 · 3dad332
1 parent 1de4ceb
commit 3dad332
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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