From 924c3fc6fd8a127a41e840ffb50396acbf7f21f3 Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Sun, 23 May 2021 22:32:43 -0700 Subject: [PATCH] fix shuffle --- samplelib/SampleProcessor.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py index 5fa854c..3b657d7 100644 --- a/samplelib/SampleProcessor.py +++ b/samplelib/SampleProcessor.py @@ -217,7 +217,9 @@ class SampleProcessor(object): ct_sample_bgr = ct_sample.load_bgr() img = imagelib.color_transfer (ct_mode, img, cv2.resize( ct_sample_bgr, (resolution,resolution), interpolation=cv2.INTER_LINEAR ) ) - randomization_order = np.random.shuffle(['blur', 'noise', 'jpeg', 'down']) + + randomization_order = ['blur', 'noise', 'jpeg', 'down'] + np.random.shuffle(randomization_order) for random_distortion in randomization_order: # Apply random blur if random_distortion == 'blur' and random_blur: