diff --git a/PaddleCV/ocr_recognition/data_reader.py b/PaddleCV/ocr_recognition/data_reader.py index f1b529391d9fb2ba8f2c43ce2257b29ac971374b..6e42e2af4fedd87ac65cb56a9461aba9b98041d4 100644 --- a/PaddleCV/ocr_recognition/data_reader.py +++ b/PaddleCV/ocr_recognition/data_reader.py @@ -212,8 +212,7 @@ class DataGenerator(object): img = img.resize((img.size[0], DATA_SHAPE[1])) # resize height img = np.array(img) - 127.5 img = img[np.newaxis, ...] - label = [int(c) for c in line.split(' ')[3].split(',')] - yield img, label + yield img, [[0]] if img_label_list is not None: lines = []