From c0ccc57ea8224dd790149b155770a600cd801b3c Mon Sep 17 00:00:00 2001
From: deeppomf <deeppomf@gmail.com>
Date: Sun, 25 Feb 2018 09:47:43 -0500
Subject: [PATCH] better threshold value for mask

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 767ab94..86d5be3 100644
--- a/train.py
+++ b/train.py
@@ -137,7 +137,7 @@ def get_points():
         	#rotate random amount between 0 and 90 degrees
         	m = scipy.ndimage.rotate(m, np.random.random()*90, reshape = False)
         	#set all elements greater than 0 to 1
-        	m[m > 0] = 1
+        	m[m > 0.5] = 1
 
         mask.append(m)