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

Foram adicionadas mensagems no proprio log do pynovisao

parent cdb23adf
......@@ -36,7 +36,7 @@ from util.x11_colors import X11Colors
import functools as ft
import multiprocessing
from multiprocessing import Process, Manager
import threading
import threading as th
from tqdm import tqdm
class Act(object):
"""Store all actions of Pynovisao."""
......@@ -1130,10 +1130,11 @@ class Act(object):
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)
imgog=cv2.imread(arquivo)
self.tk.add_image(imgog, file)
LinhaOg,ColunaOg,_=imgog.shape
# 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)
for pseg, _,filesIO in os.walk(os.path.abspath(pathS)):
pasta=pseg#pega o nome da pasta (classe do segmento)
......@@ -1145,10 +1146,9 @@ class Act(object):
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)
imgseg=cv2.imread(arquivo)
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)
......@@ -1157,15 +1157,15 @@ class Act(object):
#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))
self.tk.append_log("\n Annotation of %s found", fileIO)
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))
self.tk.append_log("\n Saving .XML")
self.tk.append_log(" Time elapsed: %0.3f seconds", tottime)
writer.save('{}.{}'.format(file,'xml'))
#gera o xml com as medidas
current_path=os.getcwd()
......@@ -1176,12 +1176,20 @@ class Act(object):
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")
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)
def config_xml(self):
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):
config_xml.destroy()
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