mirror/DeepCreamPy
1
0
Fork 0
mirror of https://github.com/Deepshift/DeepCreamPy.git synced 2024-11-29 05:10:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

better threshold value for mask

Browse Source
This commit is contained in:
deeppomf 2018-02-25 09:47:43 -05:00
parent fe86070b74
commit c0ccc57ea8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -137,7 +137,7 @@ def get_points():
#rotate random amount between 0 and 90 degrees #rotate random amount between 0 and 90 degrees
m = scipy.ndimage.rotate(m, np.random.random()*90, reshape = False) m = scipy.ndimage.rotate(m, np.random.random()*90, reshape = False)
#set all elements greater than 0 to 1 #set all elements greater than 0 to 1
m[m > 0] = 1 m[m > 0.5] = 1
mask.append(m) mask.append(m)