def selectMode(self, configDict, automatic):
mode = ""
if automatic == True:
mode = getattr(self, "mode")
else:
while not mode:
mode = input(
"Enter mode. \'prepare\' , \'classify\' , \'train\' or \'predict\'\n"
)
if mode == "classify" or mode == "train" or mode == "predict" or mode == "prepare":
print("Correct mode chosen.")
else:
print("Wrong mode!")
mode = ""
if mode == "classify":
from classify import Classify
classify = Classify(**dict(configDict["Classify"]))
classify.classifyMultipleImages()
elif mode == "train":
from trainModel import TrainModel
train = TrainModel(**dict(configDict["Train"]))
train.createModel()
elif mode == "predict":
from predictService import PredictService
predict = PredictService(**dict(configDict["Predict"]))
predict.downloadImage()
predict.predict()
elif mode == "prepare":
args = {
"pathToConfig": configDict["Prepare"]["configPath"],
}
self.GlobalServices(**args).prepareFolders(
input(
"Enter root folder name. (Contains default folder structure.)\n"
))
