Fixed python2 integer division

apache · Nov 13, 2019 · 50abb88 · 50abb88
1 parent 676f840
commit 50abb88
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/mxnet/image/image.py b/python/mxnet/image/image.py
@@ -676,8 +676,8 @@ def imrotate(src, rotation_degrees, zoom_in=False, zoom_out=False):
     _, _, h, w = src.shape
 
     # Generate a grid centered at the center of the image
-    hscale = ((h - 1) / 2)
-    wscale = ((w - 1) / 2)
+    hscale = (float(h - 1) / 2)
+    wscale = (float(w - 1) / 2)
     h_matrix = (
         nd.repeat(nd.arange(h, ctx=src.context).astype('float32').reshape(h, 1), w, axis=1) - hscale
     ).expand_dims(axis=0)