update; type params

oneoffcoder · Jan 22, 2020 · 601f7ba · 601f7ba
1 parent b81701e
commit 601f7ba
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 9 deletions.
diff --git a/dl-transfer/.dockerignore b/dl-transfer/.dockerignore
@@ -0,0 +1 @@
+gvp/
diff --git a/dl-transfer/.gitignore b/dl-transfer/.gitignore
@@ -0,0 +1 @@
+gvp/
diff --git a/dl-transfer/app.py b/dl-transfer/app.py
@@ -223,14 +223,14 @@ def parse_args(args):
     parser = argparse.ArgumentParser('Neural Style Transfer',
         epilog='One-Off Coder http://www.oneoffcoder.com')
 
-    parser.add_argument('--seed', help='Seed', default=37, required=False)
-    parser.add_argument('--steps', help='Number of steps', default=600, required=False)
-    parser.add_argument('--sweight', help='Style weight', default=1000000, required=False)
-    parser.add_argument('--cweight', help='Content weight', default=1, required=False)
-
-    parser.add_argument('-s', '--style', help='Style image path', required=True)
-    parser.add_argument('-c', '--content', help='Content image path', required=True)
-    parser.add_argument('-o', '--output', help='Output image path', default='./image/output.jpg', required=False)
+    parser.add_argument('--seed', help='Seed', type=int, default=37, required=False)
+    parser.add_argument('--steps', help='Number of steps', type=int, default=600, required=False)
+    parser.add_argument('--sweight', help='Style weight', type=int, default=1000000, required=False)
+    parser.add_argument('--cweight', help='Content weight', type=int, default=1, required=False)
+
+    parser.add_argument('-s', '--style', help='Style image path', type=str, required=True)
+    parser.add_argument('-c', '--content', help='Content image path', type=str, required=True)
+    parser.add_argument('-o', '--output', help='Output image path', type=str, default='./image/output.jpg', required=False)
 
     return parser.parse_args(args)
 

diff --git a/dl-transfer/deploy.sh b/dl-transfer/deploy.sh
@@ -2,7 +2,7 @@
 
 ORGANIZATION=oneoffcoder
 REPOSITORY=dl-transfer
-VERSION=0.0.2
+VERSION=0.0.3
 IMAGEID=${REPOSITORY}:local
 
 echo ${IMAGEID}