def morph(): mf = MorphologyFeatureExtractor() mf.setThresholdOperation(myBinaryFunc) a = np.array(mf.extract(watchman.imgbank[0])) if len(watchman.imgbank) > 1: b = np.array(mf.extract(watchman.imgbank[1])) else: b = np.array(mf.extract(watchman.imgbank[0])) AandB = np.sum(np.square(a-b)) return AandB
def morph(): mf = MorphologyFeatureExtractor() mf.setThresholdOperation(myBinaryFunc) a = np.array(mf.extract(watchman.imgbank[0])) if len(watchman.imgbank) > 1: b = np.array(mf.extract(watchman.imgbank[1])) else: b = np.array(mf.extract(watchman.imgbank[0])) AandB = np.sum(np.square(a - b)) return AandB
def feature_extractor(): #Abrir arquivo que vai receber as features dos arquivos de imagens file = "image_features_train_20_10_1_Novo4.arff" text_file = open(file, "a") trainPath = "C:/Python27/ARP/Dataset/train5" dirList = os.listdir(trainPath) edge = EdgeHistogramFeatureExtractor(20) for dirName in dirList: fileList = os.listdir(trainPath + '/' + dirName) maxsize = len(fileList) for i in range(0, maxsize): #print fileList[i] filePATH = (trainPath + '/' + dirName + '/' + fileList[i]) img = Image(filePATH) #Feature - Edge histogram edge_vecs = edge.extract(img) edge_fields = edge.getFieldNames() for y in range(0, edge.getNumFields()): #print edge_fields[y], '=', edge_vecs[y] text_file.write(str(edge_vecs[y]) + ",") #Feature - Hue hue = HueHistogramFeatureExtractor(10) hue_vecs = hue.extract(img) hue_fields = hue.getFieldNames() for i in range(0, hue.getNumFields()): # #print hue_fields[i], '=', hue_vecs[i] text_file.write(str(hue_vecs[i]) + ",") Morph1 = MorphologyFeatureExtractor() Morph2 = Morph1.extract(img) text_file.write(str(Morph2) + ",") text_file.write(dirName + "\n") text_file.close()
for i in range(0, hue.getNumFields()): print hue_fields[i], '=', hue_vecs[i] text_file.write(str(hue_vecs[i]) + ",") #Feature - Haar #haar = HaarLikeFeatureExtractor(fname='haar.txt') #haar_vecs = haar.extract(img) #haar_fields = haar.getFieldNames() #for i in range(0, len(haar_fields)): # print haar_fields[i], '=', haar_vecs[i] # text_file.write(str(haar_vecs[i]) + ",") text_file.write("revolver") text_file.close() """ #Criando o header do arquivo ARFF de features #var1 = gerador_Header() #Gerando e gravando as features no arquivo ARFF #var2 = feature_extractor() """ #classes = ['pistola_gray','revolver_gray',] """ img = Image('revolver.jpg') feature = MorphologyFeatureExtractor() array = feature.extract(img) print (array) #text_file.write("revolver") #text_file.close()