Skip to content

Commit ada0365

Browse files
committed
Use `choices' keyword for model argument.
1 parent e34847e commit ada0365

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

run_webcam.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434
if __name__ == '__main__':
3535

3636
parser = argparse.ArgumentParser(description='MobilePose Realtime Webcam.')
37-
parser.add_argument('--model', type=str, default='resnet18', help='mobilenetv2|resnet18|shufflenetv2|squeezenet')
37+
parser.add_argument('--model', type=str, default='resnet18', choices=['mobilenetv2', 'resnet18', 'shufflenetv2', 'squeezenet'])
3838
parser.add_argument('--inp_dim', type=int, default=224, help='input size')
3939
parser.add_argument('--camera', type=int, default=0)
4040

0 commit comments

Comments
 (0)