mirrors
/
mmocr
mirror of https://github.com/open-mmlab/mmocr.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix issue#127: changed save_path to out_file (#141)

pull/142/head
quincylin1 2021-04-29 15:46:46 +08:00 committed by GitHub
parent e29b5b8abe
commit 40f10cc996
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
demo/image_demo.py
View File

@ -13,7 +13,7 @@ def main():
parser.add_argument('img', help='Image file.')
parser.add_argument('config', help='Config file.')
parser.add_argument('checkpoint', help='Checkpoint file.')
parser.add_argument('save_path', help='Path to save visualized image.')
parser.add_argument('out_file', help='Path to save visualized image.')
parser.add_argument(
'--device', default='cuda:0', help='Device used for inference.')
parser.add_argument(
@ -33,9 +33,13 @@ def main():
print(f'result: {result}')
# show the results
img = model.show_result(args.img, result, out_file=None, show=False)
img = model.show_result(
args.img, result, out_file=args.out_file, show=False)
mmcv.imwrite(img, args.save_path)
if img is None:
img = mmcv.imread(args.img)
mmcv.imwrite(img, args.out_file)
if args.imshow:
mmcv.imshow(img, 'predicted results')