diff --git a/src/dataloader_iam.py b/src/dataloader_iam.py index 5aa8db7c7ca62fd6338bb7f120b97a6158c400d2..e0addeb1f3fc3a252f07a995b4a641b89afdf5e4 100644 --- a/src/dataloader_iam.py +++ b/src/dataloader_iam.py @@ -40,11 +40,12 @@ class DataLoaderIAM: chars = set() bad_samples_reference = ['a01-117-05-02', 'r06-022-03-05'] # known broken images in IAM dataset for line in f: - # ignore comment line + # ignore empty and comment lines + line = line.strip() if not line or line[0] == '#': continue - line_split = line.strip().split(' ') + line_split = line.split(' ') assert len(line_split) >= 9 # filename: part1-part2-part3 --> part1/part1-part2/part1-part2-part3.png