提交 11d6814f 编写于 作者: 文幕地方's avatar 文幕地方

rm _

上级 9131c4a7
...@@ -63,7 +63,7 @@ def infer(args): ...@@ -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)
......
...@@ -250,7 +250,7 @@ def infer(args): ...@@ -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)
......
...@@ -137,7 +137,7 @@ if __name__ == "__main__": ...@@ -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)
......
...@@ -120,7 +120,7 @@ if __name__ == "__main__": ...@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册