From 8577a9e8c6a4c72a0dfd3228dd0b68c5ba795c72 Mon Sep 17 00:00:00 2001 From: Harald Scheidl <harald@newpc.com> Date: Thu, 13 Oct 2022 16:27:29 +0200 Subject: [PATCH] ignore empty lines and comment lines in IAM dataset file --- src/dataloader_iam.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/dataloader_iam.py b/src/dataloader_iam.py index 5aa8db7..e0addeb 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 -- GitLab