diff --git a/modules/deeplearning/unet_l1b_l2.py b/modules/deeplearning/unet_l1b_l2.py index 0afb47289aef481d5049c8bb68e0aa8eb1e65267..764df959cccbef554f61ba3c9e2b9254f72db1e4 100644 --- a/modules/deeplearning/unet_l1b_l2.py +++ b/modules/deeplearning/unet_l1b_l2.py @@ -901,6 +901,11 @@ class UNET: train_label_files = glob.glob(directory+'label_train*.npy') valid_label_files = glob.glob(directory+'label_valid*.npy') + train_data_files.sort() + valid_data_files.sort() + train_label_files.sort() + valid_label_files.sort() + self.setup_pipeline_files(train_data_files, train_label_files, valid_data_files, valid_label_files) self.build_model() self.build_training()