From 920460e3cbdc8bc43ecb2269ac4a5b91efc1ad10 Mon Sep 17 00:00:00 2001
From: SlimRG <39348033+SlimRG@users.noreply.github.com>
Date: Thu, 11 Oct 2018 20:31:55 +0300
Subject: [PATCH] TypeError: 'float' object cannot be interpreted as an integer

Look into Issues (https://github.com/jocicmarko/ultrasound-nerve-segmentation/issues/79)
---
 data.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/data.py b/data.py
index 418c809..de39110 100644
--- 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)