From 278969b63fd92c2bdbc02e6634e2c2b83879cbcc Mon Sep 17 00:00:00 2001 From: Sylwester Fraczek Date: Tue, 15 May 2018 16:06:39 +0200 Subject: [PATCH] crnn-ctc label was not found (#912) --- fluid/ocr_recognition/ctc_reader.py | 1 + 1 file changed, 1 insertion(+) diff --git a/fluid/ocr_recognition/ctc_reader.py b/fluid/ocr_recognition/ctc_reader.py index aa7c4edd..ae8912b3 100644 --- a/fluid/ocr_recognition/ctc_reader.py +++ b/fluid/ocr_recognition/ctc_reader.py @@ -136,6 +136,7 @@ class DataGenerator(object): img = Image.open(img_path).convert('L') img = np.array(img) - 127.5 img = img[np.newaxis, ...] + label = [int(c) for c in line.split(' ')[3].split(',')] yield img, label else: while True: -- GitLab