From bb1e4b9ea1819c72f5cc8b11bb5032cf5782f79a Mon Sep 17 00:00:00 2001 From: whs Date: Wed, 6 Nov 2019 11:23:11 +0800 Subject: [PATCH] Remove label from infer reader. (#3882) --- PaddleCV/ocr_recognition/data_reader.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/PaddleCV/ocr_recognition/data_reader.py b/PaddleCV/ocr_recognition/data_reader.py index f1b52939..6e42e2af 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 = [] -- GitLab