Fix: dataset names error handling
This commit is contained in:
parent
1e7d5de144
commit
6eeb3cca53
|
@ -56,7 +56,7 @@ class DatasetFFDNet(data.Dataset):
|
||||||
H_file, L_file = H_path.split('/')[-1], L_path.split('/')[-1]
|
H_file, L_file = H_path.split('/')[-1], L_path.split('/')[-1]
|
||||||
H_name, L_name = H_file.split('.')[0], L_file.split('.')[0]
|
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)
|
img_H = util.imread_uint(H_path, self.n_channels_datasetload)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue