diff options
Diffstat (limited to 'imagocli.py')
| -rwxr-xr-x | imagocli.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/imagocli.py b/imagocli.py index 5f14380..d660195 100755 --- a/imagocli.py +++ b/imagocli.py @@ -14,12 +14,15 @@ if __name__ == "__main__": if sys.argv[1] == "-e": if sys.argv[2] == "montecarlo": decisionAlgorithm = DecisionAlgorithms.MONTECARLO - if sys.argv[2] == "keras": + elif sys.argv[2] == "keras": decisionAlgorithm = DecisionAlgorithms.KERAS - if sys.argv[2] == "dense": + elif sys.argv[2] == "dense": decisionAlgorithm = DecisionAlgorithms.DENSE - if sys.argv[2] == "conv": + elif sys.argv[2] == "conv": decisionAlgorithm = DecisionAlgorithms.CONV + else: + print('Invalid algorithm "%s"' % sys.argv[2]) + sys.exit(1) if decisionAlgorithm is None: io = ImagoIO() |