Commit 53374536 authored by Fábio Prestes's avatar Fábio Prestes
Browse files

Foram adicionadas mensagems no proprio log do pynovisao

parent cdb23adf
...@@ -36,7 +36,7 @@ from util.x11_colors import X11Colors ...@@ -36,7 +36,7 @@ from util.x11_colors import X11Colors
import functools as ft import functools as ft
import multiprocessing import multiprocessing
from multiprocessing import Process, Manager from multiprocessing import Process, Manager
import threading import threading as th
from tqdm import tqdm from tqdm import tqdm
class Act(object): class Act(object):
"""Store all actions of Pynovisao.""" """Store all actions of Pynovisao."""
...@@ -1130,10 +1130,11 @@ class Act(object): ...@@ -1130,10 +1130,11 @@ class Act(object):
if(file.endswith('.jpeg') or file.endswith('.tif') or file.endswith('.jpg') or file.endswith('.png')):##(-f) 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 #leitura dos arquivos de imagem na pasta original
arquivo=os.path.join(pog, file) arquivo=os.path.join(pog, file)
imgog=cv2.imread(arquivo,-1) imgog=cv2.imread(arquivo)
self.tk.add_image(imgog, file)
LinhaOg,ColunaOg,_=imgog.shape LinhaOg,ColunaOg,_=imgog.shape
# Writer(path, width, height) para iniciar o geramento do xml # Writer(path, width, height) para iniciar o geramento do xml
print("criando xml para a imagem {}".format(file)) self.tk.write_log(" Creating XML file for image %s", file)
writer = wr(pog, LinhaOg, ColunaOg) writer = wr(pog, LinhaOg, ColunaOg)
for pseg, _,filesIO in os.walk(os.path.abspath(pathS)): for pseg, _,filesIO in os.walk(os.path.abspath(pathS)):
pasta=pseg#pega o nome da pasta (classe do segmento) pasta=pseg#pega o nome da pasta (classe do segmento)
...@@ -1145,10 +1146,9 @@ class Act(object): ...@@ -1145,10 +1146,9 @@ class Act(object):
if (fileIO.startswith(file)): #verifica se eh o segmento desta imagem ou nao. if (fileIO.startswith(file)): #verifica se eh o segmento desta imagem ou nao.
#leitura dos segmentos em seus diretorios #leitura dos segmentos em seus diretorios
arquivo=os.path.join(pseg,fileIO) arquivo=os.path.join(pseg,fileIO)
imgseg=cv2.imread(arquivo,-1) imgseg=cv2.imread(arquivo)
LinhaSeg,ColunaSeg,_=imgseg.shape LinhaSeg,ColunaSeg,_=imgseg.shape
l=0 l=0
print("buscando encaixe do superpixel {}".format(fileIO))
#arraste do superpixel sobre a imagem original #arraste do superpixel sobre a imagem original
starttime = time.time() starttime = time.time()
l,c,i,j,igual=arrasteSeg(imgog, imgseg, LinhaOg, ColunaOg, LinhaSeg, ColunaSeg) l,c,i,j,igual=arrasteSeg(imgog, imgseg, LinhaOg, ColunaOg, LinhaSeg, ColunaSeg)
...@@ -1157,15 +1157,15 @@ class Act(object): ...@@ -1157,15 +1157,15 @@ class Act(object):
#tudo isso pra terminar o loop desse segmento #tudo isso pra terminar o loop desse segmento
if not(fileIO in arquivos): #Se o segmento atual ja foi visto if not(fileIO in arquivos): #Se o segmento atual ja foi visto
# ::addObject(name, xmin, ymin, xmax, ymax) # ::addObject(name, xmin, ymin, xmax, ymax)
print("adicionando anotacao, superpixel {} encontrado em {}".format(fileIO,file)) self.tk.append_log("\n Annotation of %s found", fileIO)
tottime = tottime + time.time() - starttime tottime = tottime + time.time() - starttime
arquivos.append(fileIO) #Adiciona nos segmentos ja vistos arquivos.append(fileIO) #Adiciona nos segmentos ja vistos
writer.addObject(pasta, c, l, c + ColunaSeg, l + LinhaSeg) #adiciona o objeto no xml writer.addObject(pasta, c, l, c + ColunaSeg, l + LinhaSeg) #adiciona o objeto no xml
l=LinhaOg+1 l=LinhaOg+1
c=ColunaOg+1 c=ColunaOg+1
#save(path) termina o salvamento do xml #save(path) termina o salvamento do xml
print("salvando xml") self.tk.append_log("\n Saving .XML")
print("Tempo decorrido em segundos: {}".format(tottime)) self.tk.append_log(" Time elapsed: %0.3f seconds", tottime)
writer.save('{}.{}'.format(file,'xml')) writer.save('{}.{}'.format(file,'xml'))
#gera o xml com as medidas #gera o xml com as medidas
current_path=os.getcwd() current_path=os.getcwd()
...@@ -1176,12 +1176,20 @@ class Act(object): ...@@ -1176,12 +1176,20 @@ class Act(object):
if(not os.path.exists(current_path)): if(not os.path.exists(current_path)):
os.makedirs(current_path) os.makedirs(current_path)
shutil.move(os.getcwd()+"/{}.xml".format(file),current_path+"/{}.xml".format(file)) shutil.move(os.getcwd()+"/{}.xml".format(file),current_path+"/{}.xml".format(file))
print("Processo finalizado") img = np.zeros([100,100,3],dtype=np.uint8)
img.fill(255)
self.tk.add_image(img, "end")
self.tk.write_log("Process finished")
self.tk.write_log("Started mass annotation process, it may take a while...")
achaImg(self._seg_folder,self._img_folder) achaImg(self._seg_folder,self._img_folder)
def config_xml(self): def config_xml(self):
config_xml = Toplevel() config_xml = Toplevel()
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"
def close_config_xml(config_xml): def close_config_xml(config_xml):
config_xml.destroy() config_xml.destroy()
def config_xml_directory(Button, E): def config_xml_directory(Button, E):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment