better threshold value for mask

2025-03-25 10:43:28 +00:00 · 2018-02-25 09:47:43 -05:00 · 2018-02-25 09:47:43 -05:00 · c0ccc57ea8
commit c0ccc57ea8
parent fe86070b74
1 changed files with 1 additions and 1 deletions
--- 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)