Skip to content

Commit 7f42009

Browse files
committed
fix get reader
1 parent d4c29dd commit 7f42009

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tools/paddle_infer.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,11 +63,13 @@ def init_predictor(args):
6363

6464
def create_data_loader(args):
6565
data_dir = args.data_dir
66-
reader_file = args.reader_file.split(".")[0]
66+
reader_path, reader_file = os.path.split(args.reader_file)
67+
reader_file, extension = os.path.splitext(reader_file)
6768
batchsize = args.batchsize
6869
place = args.place
6970
file_list = [os.path.join(data_dir, x) for x in os.listdir(data_dir)]
70-
sys.path.append(os.path.abspath("."))
71+
sys.path.append(reader_path)
72+
#sys.path.append(os.path.abspath("."))
7173
reader_class = import_module(reader_file)
7274
dataset = reader_class.RecDataset(file_list, config=None)
7375
loader = DataLoader(

0 commit comments

Comments
 (0)