def calcula_distancia_imagens_semelhantes_distintas(self,img_consulta,num_imagens_distintas): imgs = im.imagems() linha = [] print(img_consulta) for img_semelhante in imgs.get_imagens_semelhantes(img_consulta): img_semelhante = os.path.basename(img_semelhante).replace('.jpg','') self.img_consulta = img_consulta self.img_base = img_semelhante linha.append(self.calcula_distancia_imagens(1)) for img_distinta in imgs.get_imagens_distintas(img_consulta,num_imagens_distintas): img_distinta = os.path.basename(img_distinta).replace('.jpg','') self.img_consulta = img_consulta self.img_base = img_distinta linha.append(self.calcula_distancia_imagens(0)) return linha
def coleta_descritores(): descritores = desc.descritor() imagens = img.imagems() imagens.path_imagens_consulta = path_imagens_consulta imagens.path_imagens_base = path_imagens_base descritores.path_descritores = path_descritores arquivo = open(str(file_base_dados), "w") arquivo.write("imgs,acc,bic,ccv,eoac,gch,las,lch,qcch,sid,unser,class\n") for imgs_consulta in imagens.get_imagens_consulta(): for imgs_base in imagens.get_imagens_base(): descritores.img_consulta = imgs_consulta descritores.img_base = imgs_base linha = descritores.calcula_distancia_imagens_consulta_base() linha = str(linha).replace("[", "").replace("]", "").replace("'", "") print("\n" + linha) arquivo.write(linha) arquivo.write("\n")
def calcula_distancia_imagens_semelhantes_distintas( self, img_consulta, num_imagens_distintas): imgs = im.imagems() linha = [] print(img_consulta) for img_semelhante in imgs.get_imagens_semelhantes(img_consulta): img_semelhante = os.path.basename(img_semelhante).replace( '.jpg', '') self.img_consulta = img_consulta self.img_base = img_semelhante linha.append(self.calcula_distancia_imagens(1)) for img_distinta in imgs.get_imagens_distintas(img_consulta, num_imagens_distintas): img_distinta = os.path.basename(img_distinta).replace('.jpg', '') self.img_consulta = img_consulta self.img_base = img_distinta linha.append(self.calcula_distancia_imagens(0)) return linha