提交 a55df348 编写于 作者: S shippingwang

fix test data path

上级 2c65b659
...@@ -168,7 +168,6 @@ Inference is used to get prediction score or image features based on trained mod ...@@ -168,7 +168,6 @@ Inference is used to get prediction score or image features based on trained mod
``` ```
python infer.py \ python infer.py \
--model=SE_ResNeXt50_32x4d \ --model=SE_ResNeXt50_32x4d \
--batch_size=32 \
--class_dim=1000 \ --class_dim=1000 \
--image_shape=3,224,224 \ --image_shape=3,224,224 \
--with_mem_opt=True \ --with_mem_opt=True \
......
...@@ -164,7 +164,6 @@ Testbatch 80,loss 0.0969972759485, acc1 1.0,acc5 1.0,time 0.41 sec ...@@ -164,7 +164,6 @@ Testbatch 80,loss 0.0969972759485, acc1 1.0,acc5 1.0,time 0.41 sec
``` ```
python infer.py \ python infer.py \
--model=SE_ResNeXt50_32x4d \ --model=SE_ResNeXt50_32x4d \
--batch_size=32 \
--class_dim=1000 \ --class_dim=1000 \
--image_shape=3,224,224 \ --image_shape=3,224,224 \
--with_mem_opt=True \ --with_mem_opt=True \
......
...@@ -156,11 +156,13 @@ def _reader_creator(file_list, ...@@ -156,11 +156,13 @@ def _reader_creator(file_list,
for line in lines: for line in lines:
if mode == 'train' or mode == 'val': if mode == 'train' or mode == 'val':
img_path, label = line.split() 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) img_path = os.path.join(data_dir, img_path)
yield img_path, int(label) yield img_path, int(label)
elif mode == 'test': 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] yield [img_path]
mapper = functools.partial( mapper = functools.partial(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册