From 05c90708a3c9e2bff6b5f810be5e81ab9882dad2 Mon Sep 17 00:00:00 2001 From: MaxwellRebo Date: Fri, 1 Dec 2017 19:46:18 +0400 Subject: [PATCH 1/2] image path arg --- demo.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/demo.py b/demo.py index c78e83ff..7d6c0ebf 100644 --- a/demo.py +++ b/demo.py @@ -5,10 +5,17 @@ from PIL import Image import models.crnn as crnn +import argparse + +parser = argparse.ArgumentParser() +parser.add_argument("--image", help="path of the image to test on") +args = parser.parse_args() + +print args model_path = './data/crnn.pth' -img_path = './data/demo.png' +img_path = args.image or './data/demo.png' alphabet = '0123456789abcdefghijklmnopqrstuvwxyz' model = crnn.CRNN(32, 1, 37, 256) From e8ab66734540e5cc3e9b8a5b9576e03ec1e75e47 Mon Sep 17 00:00:00 2001 From: MaxwellRebo Date: Fri, 1 Dec 2017 19:47:09 +0400 Subject: [PATCH 2/2] don't print args --- demo.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/demo.py b/demo.py index 7d6c0ebf..37093ba6 100644 --- a/demo.py +++ b/demo.py @@ -11,9 +11,6 @@ parser.add_argument("--image", help="path of the image to test on") args = parser.parse_args() -print args - - model_path = './data/crnn.pth' img_path = args.image or './data/demo.png' alphabet = '0123456789abcdefghijklmnopqrstuvwxyz'