Update for python 3

mmreza79 · Oct 29, 2017 · 4195f10 · 4195f10
1 parent 201e525
commit 4195f10
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Colorful/src/utils/general_utils.py b/Colorful/src/utils/general_utils.py
@@ -69,7 +69,7 @@ def plot_batch(color_model, q_ab, X_batch_black, X_batch_color, batch_size, h, w
         list_img.append(arr)
 
     plt.figure(figsize=(20,20))
-    list_img = [np.concatenate(list_img[4 * i: 4 * (i + 1)], axis=2) for i in range(len(list_img) / 4)]
+    list_img = [np.concatenate(list_img[4 * i: 4 * (i + 1)], axis=2) for i in range(len(list_img) // 4)]
     arr = np.concatenate(list_img, axis=1)
     plt.imshow(arr.transpose(1,2,0))
     ax = plt.gca()
@@ -111,7 +111,7 @@ def plot_batch_eval(color_model, q_ab, X_batch_black, X_batch_color, batch_size,
         list_img.append(arr)
 
     plt.figure(figsize=(20,20))
-    list_img = [np.concatenate(list_img[4 * i: 4 * (i + 1)], axis=2) for i in range(len(list_img) / 4)]
+    list_img = [np.concatenate(list_img[4 * i: 4 * (i + 1)], axis=2) for i in range(len(list_img) // 4)]
     arr = np.concatenate(list_img, axis=1)
     plt.imshow(arr.transpose(1,2,0))
     ax = plt.gca()