yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-10-31 17:59:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

rm _

Browse Source
This commit is contained in:
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( save_img_path = os.path.join(
args.output_dir, args.output_dir,
os.path.splitext(os.path.basename(image_path))[0] + "_re.jpg") 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)) idx, len(eval_dataloader), save_img_path))
with paddle.no_grad(): with paddle.no_grad():
outputs = model(**batch) 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): for idx, img_path in enumerate(infer_imgs):
save_img_path = os.path.join(args.output_dir, save_img_path = os.path.join(args.output_dir,
os.path.basename(img_path)) os.path.basename(img_path))
print("process: [{}/{}], save_result to {}".format( print("process: [{}/{}], save result to {}".format(
idx, len(infer_imgs), save_img_path)) idx, len(infer_imgs), save_img_path))
img = cv2.imread(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( save_img_path = os.path.join(
args.output_dir, args.output_dir,
os.path.splitext(os.path.basename(img_path))[0] + "_ser.jpg") 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)) idx, len(infer_imgs), save_img_path))
img = cv2.imread(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( save_img_path = os.path.join(
args.output_dir, args.output_dir,
os.path.splitext(os.path.basename(img_path))[0] + "_re.jpg") 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)) idx, len(infer_imgs), save_img_path))
img = cv2.imread(img_path) img = cv2.imread(img_path)