Wiring in the cli parameters.

Author: rigolepe

README.md

@@ -75,7 +75,7 @@ version and converted on the fly.
 Image demo runs inference on an input folder of images and outputs those images with the keypoints and skeleton 
 overlayed.
 
-`python image_demo.py --model 101 --image_dir ./images --output_dir ./output`
+`python image_demo.py --model resnet50 --stride 32 --image_dir ./images --output_dir ./output`
 
 A folder of suitable test images can be downloaded by first running the `get_test_images.py` script.
 

benchmark.py

@@ -7,7 +7,10 @@
 
 
 parser = argparse.ArgumentParser()
-parser.add_argument('--model', type=int, default=101)
+parser.add_argument('--model', type=str, default='resnet50')  # mobilenet resnet50
+parser.add_argument('--stride', type=int, default=16)  # 8, 16, 32 (max 16 for mobilenet)
+parser.add_argument('--quant_bytes', type=int, default=4)  # 4 = float
+parser.add_argument('--multiplier', type=float, default=1.0)  # only for mobilenet
 parser.add_argument('--image_dir', type=str, default='./images')
 parser.add_argument('--num_images', type=int, default=1000)
 args = parser.parse_args()
@@ -18,10 +21,10 @@ def main():
     print('Tensorflow version: %s' % tf.__version__)
     assert tf.__version__.startswith('2.'), "Tensorflow version 2.x must be used!"
 
-    model = 'resnet50'  # mobilenet resnet50
-    stride = 32  # 8, 16, 32
-    quant_bytes = 4  # float
-    multiplier = 1.0  # only for mobilenet
+    model = args.model  # mobilenet resnet50
+    stride = args.stride  # 8, 16, 32 (max 16 for mobilenet)
+    quant_bytes = args.quant_bytes  # float
+    multiplier = args.multiplier  # only for mobilenet
 
     posenet = load_model(model, stride, quant_bytes, multiplier)
 

image_demo.py

@@ -6,8 +6,10 @@
 from posenet.posenet_factory import load_model
 
 parser = argparse.ArgumentParser()
-parser.add_argument('--model', type=int, default=101)
-parser.add_argument('--scale_factor', type=float, default=1.0)
+parser.add_argument('--model', type=str, default='resnet50')  # mobilenet resnet50
+parser.add_argument('--stride', type=int, default=16)  # 8, 16, 32 (max 16 for mobilenet)
+parser.add_argument('--quant_bytes', type=int, default=4)  # 4 = float
+parser.add_argument('--multiplier', type=float, default=1.0)  # only for mobilenet
 parser.add_argument('--notxt', action='store_true')
 parser.add_argument('--image_dir', type=str, default='./images')
 parser.add_argument('--output_dir', type=str, default='./output')
@@ -23,10 +25,10 @@ def main():
         if not os.path.exists(args.output_dir):
             os.makedirs(args.output_dir)
 
-    model = 'resnet50'  # mobilenet resnet50
-    stride = 32  # 8, 16, 32 (max 16 for mobilenet)
-    quant_bytes = 4  # float
-    multiplier = 1.0  # only for mobilenet
+    model = args.model  # mobilenet resnet50
+    stride = args.stride  # 8, 16, 32 (max 16 for mobilenet)
+    quant_bytes = args.quant_bytes  # float
+    multiplier = args.multiplier  # only for mobilenet
 
     posenet = load_model(model, stride, quant_bytes, multiplier)