mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

rm _

pull/4984/head
WenmuZhou 2021-12-20 14:47:43 +00:00
parent 9131c4a7ac
commit 11d6814f36
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppstructure/vqa/infer_re.py
View File

@ -63,7 +63,7 @@ def infer(args):
save_img_path = os.path.join(
args.output_dir,
os.path.splitext(os.path.basename(image_path))[0] + "_re.jpg")
logger.info("[Infer] process: {}/{}, save_result to {}".format(
logger.info("[Infer] process: {}/{}, save result to {}".format(
idx, len(eval_dataloader), save_img_path))
with paddle.no_grad():
outputs = model(**batch)

2
ppstructure/vqa/infer_ser.py
View File

@ -250,7 +250,7 @@ def infer(args):
for idx, img_path in enumerate(infer_imgs):
save_img_path = os.path.join(args.output_dir,
os.path.basename(img_path))
print("process: [{}/{}], save_result to {}".format(
print("process: [{}/{}], save result to {}".format(
idx, len(infer_imgs), save_img_path))
img = cv2.imread(img_path)

2
ppstructure/vqa/infer_ser_e2e.py
View File

@ -137,7 +137,7 @@ if __name__ == "__main__":
save_img_path = os.path.join(
args.output_dir,
os.path.splitext(os.path.basename(img_path))[0] + "_ser.jpg")
print("process: [{}/{}], save_result to {}".format(
print("process: [{}/{}], save result to {}".format(
idx, len(infer_imgs), save_img_path))
img = cv2.imread(img_path)

2
ppstructure/vqa/infer_ser_re_e2e.py
View File

@ -120,7 +120,7 @@ if __name__ == "__main__":
save_img_path = os.path.join(
args.output_dir,
os.path.splitext(os.path.basename(img_path))[0] + "_re.jpg")
print("process: [{}/{}], save_result to {}".format(
print("process: [{}/{}], save result to {}".format(
idx, len(infer_imgs), save_img_path))
img = cv2.imread(img_path)