From 6eeb3cca53582073222d6ecdff08398719e51461 Mon Sep 17 00:00:00 2001 From: Emilio Martinez Date: Mon, 26 Jun 2023 23:54:22 -0300 Subject: [PATCH] Fix: dataset names error handling --- KAIR/data/dataset_ffdnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/KAIR/data/dataset_ffdnet.py b/KAIR/data/dataset_ffdnet.py index e081d7f..a53a107 100644 --- a/KAIR/data/dataset_ffdnet.py +++ b/KAIR/data/dataset_ffdnet.py @@ -56,7 +56,7 @@ class DatasetFFDNet(data.Dataset): H_file, L_file = H_path.split('/')[-1], L_path.split('/')[-1] H_name, L_name = H_file.split('.')[0], L_file.split('.')[0] - assert H_name==L_name, 'Both high and low quality images MUST have same name' + assert H_name==L_name, f'Both high and low quality images MUST have same name.\nGot {H_name} and {L_name} respectively.' img_H = util.imread_uint(H_path, self.n_channels_datasetload)