fix bug in selection

hkchengrex · Dec 23, 2020 · 3479415 · 3479415
1 parent c354d93
commit 3479415
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/util/de_transform.py b/util/de_transform.py
@@ -3,7 +3,10 @@
 import numpy as np
 
 def get_random_structure(size):
-    choice = np.random.randint(4)
+    # The provided model is trained with 
+    #   choice = np.random.randint(4)
+    # instead, which is a bug that we fixed here
+    choice = np.random.randint(1, 5)
 
     if choice == 1:
         return cv2.getStructuringElement(cv2.MORPH_RECT, (size, size))