diff options
| author | InigoGutierrez <inigogf.95@gmail.com> | 2023-06-12 20:16:04 +0200 |
|---|---|---|
| committer | InigoGutierrez <inigogf.95@gmail.com> | 2023-06-12 20:16:04 +0200 |
| commit | d4a81490bf1396089eb3dac5955a3a8e4cb26e37 (patch) | |
| tree | f96febc7950c2742bc36f04ab13bff56851f2388 /imagocli.py | |
| parent | b08408d23186205e71dfc68634021e3236bfb45c (diff) | |
| parent | 65ac3a6b050dcb88688cdc2654b1ed6693e9a160 (diff) | |
| download | imago-d4a81490bf1396089eb3dac5955a3a8e4cb26e37.tar.gz imago-d4a81490bf1396089eb3dac5955a3a8e4cb26e37.zip | |
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() |