Skip to content

Commit

Permalink
Merge pull request #5900 from wasnot/fix/py3-division-compat
Browse files Browse the repository at this point in the history
Fix division operator for Compatibility of python 3 in classifier.py
  • Loading branch information
longjon committed Sep 6, 2017
2 parents 1de4ceb + 3dad332 commit ea455eb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/caffe/classifier.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

0 comments on commit ea455eb

Please sign in to comment.