TypeError: 'float' object cannot be interpreted as an integer

Look into Issues (jocicmarko#79)
SlimRG · Oct 11, 2018 · 920460e · 920460e
1 parent df9eb77
commit 920460e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data.py b/data.py
@@ -14,7 +14,7 @@
 def create_train_data():
     train_data_path = os.path.join(data_path, 'train')
     images = os.listdir(train_data_path)
-    total = len(images) / 2
+    total = len(images) // 2
 
     imgs = np.ndarray((total, image_rows, image_cols), dtype=np.uint8)
     imgs_mask = np.ndarray((total, image_rows, image_cols), dtype=np.uint8)