diff options
author | InigoGutierrez <inigogf.95@gmail.com> | 2022-07-01 15:40:57 +0200 |
---|---|---|
committer | InigoGutierrez <inigogf.95@gmail.com> | 2022-07-01 15:40:57 +0200 |
commit | 4cc55348c8dbb1902a1246fba66237d5c59f0349 (patch) | |
tree | ebc363dec6ae00f711be7ebd6e31530f25af1d9f /imagocli.py | |
parent | 6724aeb3ba98c1b9f042344734c2d683e79dfc64 (diff) | |
download | imago-4cc55348c8dbb1902a1246fba66237d5c59f0349.tar.gz imago-4cc55348c8dbb1902a1246fba66237d5c59f0349.zip |
Finished writing documentation.
Diffstat (limited to 'imagocli.py')
-rwxr-xr-x | imagocli.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/imagocli.py b/imagocli.py index 150587e..5f14380 100755 --- a/imagocli.py +++ b/imagocli.py @@ -4,8 +4,8 @@ import sys +from imago.data.enums import DecisionAlgorithms from imago.engine.imagoIO import ImagoIO -from imago.engine.decisionAlgorithmFactory import DecisionAlgorithms if __name__ == "__main__": @@ -16,6 +16,10 @@ if __name__ == "__main__": decisionAlgorithm = DecisionAlgorithms.MONTECARLO if sys.argv[2] == "keras": decisionAlgorithm = DecisionAlgorithms.KERAS + if sys.argv[2] == "dense": + decisionAlgorithm = DecisionAlgorithms.DENSE + if sys.argv[2] == "conv": + decisionAlgorithm = DecisionAlgorithms.CONV if decisionAlgorithm is None: io = ImagoIO() |