diff options
author | InigoGutierrez <inigogf.95@gmail.com> | 2023-06-06 20:09:08 +0200 |
---|---|---|
committer | InigoGutierrez <inigogf.95@gmail.com> | 2023-06-06 20:09:08 +0200 |
commit | 12431ce4c4ed52fcfc2180bcffdfec33e72b73ba (patch) | |
tree | 630235a3db5287604217e87126931de842ad6499 /imagocli.py | |
parent | 88fbf5f8919211cfa06116a76f42fb26ec9f2e18 (diff) | |
download | imago-12431ce4c4ed52fcfc2180bcffdfec33e72b73ba.tar.gz imago-12431ce4c4ed52fcfc2180bcffdfec33e72b73ba.zip |
Written user manuals.
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() |