From a55df3485f2b7c1f2ab124b6672975ca61d2f525 Mon Sep 17 00:00:00 2001 From: shippingwang Date: Tue, 12 Mar 2019 07:05:40 +0000 Subject: [PATCH] fix test data path --- fluid/PaddleCV/image_classification/README.md | 1 - fluid/PaddleCV/image_classification/README_cn.md | 1 - fluid/PaddleCV/image_classification/reader.py | 6 ++++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/fluid/PaddleCV/image_classification/README.md b/fluid/PaddleCV/image_classification/README.md index 3a20bdf4..86416399 100644 --- a/fluid/PaddleCV/image_classification/README.md +++ b/fluid/PaddleCV/image_classification/README.md @@ -168,7 +168,6 @@ Inference is used to get prediction score or image features based on trained mod ``` python infer.py \ --model=SE_ResNeXt50_32x4d \ - --batch_size=32 \ --class_dim=1000 \ --image_shape=3,224,224 \ --with_mem_opt=True \ diff --git a/fluid/PaddleCV/image_classification/README_cn.md b/fluid/PaddleCV/image_classification/README_cn.md index c9f553e3..803bdc99 100644 --- a/fluid/PaddleCV/image_classification/README_cn.md +++ b/fluid/PaddleCV/image_classification/README_cn.md @@ -164,7 +164,6 @@ Testbatch 80,loss 0.0969972759485, acc1 1.0,acc5 1.0,time 0.41 sec ``` python infer.py \ --model=SE_ResNeXt50_32x4d \ - --batch_size=32 \ --class_dim=1000 \ --image_shape=3,224,224 \ --with_mem_opt=True \ diff --git a/fluid/PaddleCV/image_classification/reader.py b/fluid/PaddleCV/image_classification/reader.py index d9559df0..f79d87b0 100644 --- a/fluid/PaddleCV/image_classification/reader.py +++ b/fluid/PaddleCV/image_classification/reader.py @@ -156,11 +156,13 @@ def _reader_creator(file_list, for line in lines: if mode == 'train' or mode == 'val': img_path, label = line.split() - img_path = img_path.replace("JPEG", "jpeg") + #img_path = img_path.replace("JPEG", "jpeg") img_path = os.path.join(data_dir, img_path) yield img_path, int(label) elif mode == 'test': - img_path = os.path.join(data_dir, line) + img_path, label = line.split() + #img_path = img_path.replace("JPEG", "jpeg") + img_path = os.path.join(data_dir, img_path) yield [img_path] mapper = functools.partial( -- GitLab