def __init__(self): self.stopGame = False #most predicted: self.predictions = {"rock":0, "paper":0, "scissors":0, "garb":0, "none":0} self.maximum = "" self.maxnr = 0 self.goalImg = cv.CreateImage((70,70), cv.IPL_DEPTH_8U, 1) self.predict = predictSign(20, False, 0) # takes the size of the images as input #load the model of the classification #self.problem_hand = self.predict.loadModel("knn", 1, "hands") # 1=>original images; 2=>PCA; 3=>Gabor Wavelets+original image; 4=>only Gabor Wavelets self.problem_sign = self.predict.loadModel("knn", 1, "rock") # 1=>original images; 2=>PCA; 3=>Gabor Wavelets+original image; 4=>only Gabor Wavelets
classi = classifyHands(buildOpt[str(build)],int(sizeImg)) classi.classifySVM(int(typeu), datas[dataset]) #____________________________________________________________________________________________________ elif(int(choice) == 7): dataset = raw_input('classify h => hands vs garbage; c => rock & paper & scissors ...') typeu = raw_input('choose the data 1 => original images; 2 => PCA on initial images; 3 => multiple Gabor filters + orig img; 4 => just multiple Gabor Filters...') datas = {'c':'rock', 'h':'hands'} classi = classifyHands(buildOpt[str(build)],int(sizeImg)) classi.classifyKNN(int(typeu), datas[dataset],4) #____________________________________________________________________________________________________ elif(int(choice) == 8): model = raw_input('model to built: s => svm; k => knn ...') dataset = raw_input('classify h => hands vs garbage; c => rock & paper & scissors ...') datas = {'c':'rock', 'h':'hands'} modelOpt = {'s':'svm', 'k':'knn'} typeu = raw_input('choose the data 1 => original images; 2 => PCA on initial images; 3 => multiple Gabor filters + orig img; 4 => just multiple Gabor Filters...') if(typeu == "3" or typeu == "4"): noComp = raw_input('number of components for PCA no ...') else: noComp = 0 predict = predictSign(int(sizeImg),buildOpt[str(build)], int(noComp)) predict.storeModel(modelOpt[str(model)], datas[dataset], int(typeu)) #____________________________________________________________________________________________________ elif(int(choice) == 9): skin = detectSkin() skin.findSkin()