PentHertz
/
deep-tempest
mirror of https://github.com/PentHertz/deep-tempest.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix: dataset names error handling

This commit is contained in:
Emilio Martinez 2023-06-26 23:54:22 -03:00
parent 1e7d5de144
commit 6eeb3cca53
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
KAIR/data/dataset_ffdnet.py
View File

@ -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)