README.md atualizado, e ferramenta XML adicionada.

README.md

@@ -162,6 +162,13 @@ Se der erro na reinstalação do matplotlib (depois que desinstalar), tente desi
 ```
  $ sudo apt-get remove python-matplotlib
 ```
+### Como instalar o tqdm
+
+#### Ubuntu 
+
+```
+ $ sudo pip3 install tqdm
+```
 
 ### Mais informações
     
@@ -352,10 +359,21 @@ Após a implementação de seu classificador, você deve configurá-lo no pynovi
 
 Caso você necessite de classes utilitárias, os arquivos delas devem ser criados no diretório  src/util/. Além disso, as classes utilitárias devem ser registradas como módulos no arquivo src/util/__init__.py
 
-teste
-
 Caso dê problema relacionado ao número de processos, adicione as duas váriaveis de ambiente,
 sendo que deve adicionar no número de threads que o seu processador permite:
 
 export OMP_NUM_THREADS=8
 export KMP_AFFINITY="verbose,explicit,proclist=[0,3,5,9,12,15,18,21],granularity=core"
+
+### Como utilizar as ferramentas de anotação XML
+
+Para aqueles que desejam criar arquivos XML durante o processo de segmentação, o Pynovisão agora é capaz de realizar tal tarefa.
+Após selecionar sua imagem e segmenta-la, apenas selecione os segmentos desejados, a ferramenta ira automaticamente salvar as caixas com as regiões de interesse. Quando estiver satisfeito, clique novamente em *Segmentation ->  Create .XML file* e o arquivo com as anotações em formato XML será criado na pasta *pynovisao/data/XML*, com o nome *imagem + .xml*.
+
+Caso o usuário já tenha segmentos prontos e a imagem original, também há a opção de identificar automaticamente a posição de tal segmento e criar a caixa com a região de interesse correspondente. Para utilizar esta ferramenta:
+- Selecione *XML -> Configure folders*
+- Selecione a opção para mudar o diretório desejado.
+- Na interface agora aberta, entre na pasta com as imagens originais e a pasta com todos os grupos de segmentos desejados respectivamente.
+- Clique em *Save All Directories*
+- Com os diretórios desejados agora salvos, clique em *XML -> Execute Conversion*
+- Após o processo de criação de XML, é possivel encontrá-lo em [...]/pynovisao/data/XML com o nome *imagem + .xml*.

src/pynovisao.py

@@ -72,8 +72,13 @@ class Act(object):
         self._image_name = None
         self._image_path = None
         self._xml_file = None
-        self._seg_folder = os.getcwd()
-        self._img_folder = os.getcwd()
+        current_path=os.getcwd()
+        current_path=current_path[-1::-1]
+        current_path=current_path[3::1]
+        current_path=current_path[-1::-1]
+        current_path=current_path+"data"
+        self._img_folder=current_path+"/images"
+        self._seg_folder=current_path+"/demo"
                     
         self._init_dataset(args["dataset"])
         self._init_classes(args["classes"], args["colors"])
@@ -1119,57 +1124,58 @@ class Act(object):
             arquivos = []
             #verificar arquivos em todas as pastas partindo de um diretório raiz
             for pog, _, files in os.walk(os.path.abspath(pathIO)):
-                #acha os arquivos de imagem na pasta de segmentos
-                for file in files:
-                    if(file.endswith('.jpeg') or file.endswith('.tif') or file.endswith('.jpg') or file.endswith('.png')):##(-f)
-                        #leitura dos arquivos de imagem na pasta original
-                        arquivo=os.path.join(pog, file)
-                        imgog=cv2.imread(arquivo,-1)
-                        LinhaOg,ColunaOg,_=imgog.shape
-                        # Writer(path, width, height) para iniciar o geramento do xml
-                        print("criando xml para a imagem {}".format(file))
-                        writer = wr(pog, LinhaOg, ColunaOg)
-                        for pseg, _,filesIO in os.walk(os.path.abspath(pathS)):
-                            pasta=pseg#pega o nome da pasta (classe do segmento)
-                            pasta = pasta[-1::-1]
-                            pasta = pasta[:pasta.index('/')]
-                            pasta = pasta[-1::-1]
-                            for fileIO in filesIO:
-                                if(fileIO.endswith('.jpeg')  or fileIO.endswith('.tif') or fileIO.endswith('.jpg') or fileIO.endswith('.png')):##(-f)
-                                    if (fileIO.startswith(file)): #verifica se eh o segmento desta imagem ou nao.
-                                        #leitura dos segmentos em seus diretorios
-                                        arquivo=os.path.join(pseg,fileIO)
-                                        imgseg=cv2.imread(arquivo,-1)
-                                        LinhaSeg,ColunaSeg,_=imgseg.shape
-                                        l=0
-                                        print("buscando encaixe do superpixel {}".format(fileIO))
-                                        #arraste do superpixel sobre a imagem original
-                                        starttime = time.time()
-                                        l,c,i,j,igual=arrasteSeg(imgog, imgseg, LinhaOg, ColunaOg, LinhaSeg, ColunaSeg)
-                                        if(igual): #Se o segmento esta igual a sua posicao no original
-                                            igual=False
-                                            #tudo isso pra terminar o loop desse segmento
-                                            if not(fileIO in arquivos): #Se o segmento atual ja foi visto
-                                                # ::addObject(name, xmin, ymin, xmax, ymax)
-                                                print("adicionando anotacao, superpixel {} encontrado em {}".format(fileIO,file))
-                                                tottime = tottime + time.time() - starttime
-                                                arquivos.append(fileIO) #Adiciona nos segmentos ja vistos
-                                                writer.addObject(pasta, c, l, c + ColunaSeg, l + LinhaSeg) #adiciona o objeto no xml
-                                                l=LinhaOg+1
-                                                c=ColunaOg+1
-                        #save(path) termina o salvamento do xml
-                        print("salvando xml")
-                        print("Tempo decorrido em segundos: {}".format(tottime))
-                        writer.save('{}.{}'.format(file,'xml'))
-                        #gera o xml com as medidas
-                        current_path=os.getcwd()
-                        current_path=current_path[-1::-1]
-                        current_path=current_path[3::1]
-                        current_path=current_path[-1::-1]
-                        current_path=current_path+"data/XML"
-                        if(not os.path.exists(current_path)):
-                            os.makedirs(current_path)
-                        shutil.move(os.getcwd()+"/{}.xml".format(file),current_path+"/{}.xml".format(file))
+                if (not pog.endswith('demo') and not pog.endswith('XML')):
+                    #acha os arquivos de imagem na pasta de segmentos
+                    for file in files:
+                        if(file.endswith('.jpeg') or file.endswith('.tif') or file.endswith('.jpg') or file.endswith('.png')):##(-f)
+                            #leitura dos arquivos de imagem na pasta original
+                            arquivo=os.path.join(pog, file)
+                            imgog=cv2.imread(arquivo,-1)
+                            LinhaOg,ColunaOg,_=imgog.shape
+                            # Writer(path, width, height) para iniciar o geramento do xml
+                            print("criando xml para a imagem {}".format(file))
+                            writer = wr(pog, LinhaOg, ColunaOg)
+                            for pseg, _,filesIO in os.walk(os.path.abspath(pathS)):
+                                pasta=pseg#pega o nome da pasta (classe do segmento)
+                                pasta = pasta[-1::-1]
+                                pasta = pasta[:pasta.index('/')]
+                                pasta = pasta[-1::-1]
+                                for fileIO in filesIO:
+                                    if(fileIO.endswith('.jpeg')  or fileIO.endswith('.tif') or fileIO.endswith('.jpg') or fileIO.endswith('.png')):##(-f)
+                                        if (fileIO.startswith(file)): #verifica se eh o segmento desta imagem ou nao.
+                                            #leitura dos segmentos em seus diretorios
+                                            arquivo=os.path.join(pseg,fileIO)
+                                            imgseg=cv2.imread(arquivo,-1)
+                                            LinhaSeg,ColunaSeg,_=imgseg.shape
+                                            l=0
+                                            print("buscando encaixe do superpixel {}".format(fileIO))
+                                            #arraste do superpixel sobre a imagem original
+                                            starttime = time.time()
+                                            l,c,i,j,igual=arrasteSeg(imgog, imgseg, LinhaOg, ColunaOg, LinhaSeg, ColunaSeg)
+                                            if(igual): #Se o segmento esta igual a sua posicao no original
+                                                igual=False
+                                                #tudo isso pra terminar o loop desse segmento
+                                                if not(fileIO in arquivos): #Se o segmento atual ja foi visto
+                                                    # ::addObject(name, xmin, ymin, xmax, ymax)
+                                                    print("adicionando anotacao, superpixel {} encontrado em {}".format(fileIO,file))
+                                                    tottime = tottime + time.time() - starttime
+                                                    arquivos.append(fileIO) #Adiciona nos segmentos ja vistos
+                                                    writer.addObject(pasta, c, l, c + ColunaSeg, l + LinhaSeg) #adiciona o objeto no xml
+                                                    l=LinhaOg+1
+                                                    c=ColunaOg+1
+                            #save(path) termina o salvamento do xml
+                            print("salvando xml")
+                            print("Tempo decorrido em segundos: {}".format(tottime))
+                            writer.save('{}.{}'.format(file,'xml'))
+                            #gera o xml com as medidas
+                            current_path=os.getcwd()
+                            current_path=current_path[-1::-1]
+                            current_path=current_path[3::1]
+                            current_path=current_path[-1::-1]
+                            current_path=current_path+"data/XML"
+                            if(not os.path.exists(current_path)):
+                                os.makedirs(current_path)
+                            shutil.move(os.getcwd()+"/{}.xml".format(file),current_path+"/{}.xml".format(file))
             print("Processo finalizado")
 
         achaImg(self._seg_folder,self._img_folder)
@@ -1179,7 +1185,7 @@ class Act(object):
         def close_config_xml(config_xml):
             config_xml.destroy()
         def config_xml_directory(Button, E):
-            path = self.tk.utils.ask_directory(default_dir = os.getcwd())
+            path = self.tk.utils.ask_directory(default_dir = current_path)
             if(Button == 1):
                 self._img_folder = path
                 E.delete(0,END)