From 34794158eb22ea55c5395e7cee35378a0d91b978 Mon Sep 17 00:00:00 2001 From: Rex Date: Wed, 23 Dec 2020 16:25:30 +0800 Subject: [PATCH] fix bug in selection --- util/de_transform.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/util/de_transform.py b/util/de_transform.py index d9d285b..29abcbc 100644 --- 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))