1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
from tkinter import *
from tkinter import filedialog
from PIL import Image, ImageTk
import logging
from enum import Enum
class GuiMain(object):
def __init__(self, args):
self.__master = Tk()
self.__args = args
self.__base = StringVar(self.__master, value=args["base"])
self.__predict_images = BooleanVar(self.__master, value=args["predict_images"])
self.__predict_images_top = StringVar(self.__master, value=str(args["predict_images_top"]))
self.__gui_tag = BooleanVar(self.__master, value=args["gui_tag"])
self.__open_system = BooleanVar(self.__master, value=args["open_system"])
validate_number = (self.__master.register(self.__validate_number))
Label(self.__master, text="Base directory for walking:").grid(row=0, column=0)
Entry(self.__master, textvariable=self.__base).grid(row=0, column=1, columnspan=2)
Button(self.__master, text="Browse", command=self.__browse).grid(row=0, column=3)
Checkbutton(self.__master, text="Use prediction for image tagging", variable=self.__predict_images).grid(row=1, column=0, columnspan=4, sticky=W)
Label(self.__master, text="Number of top results:").grid(row=2, column=0)
Entry(self.__master, textvariable=self.__predict_images_top, validate='all', validatecommand=(validate_number, '%P')).grid(row=2, column=1, columnspan=1)
Checkbutton(self.__master, text="Show GUI for tagging", variable=self.__gui_tag).grid(row=3, column=0, columnspan=4, sticky=W)
Checkbutton(self.__master, text="Open all files with system default", variable=self.__open_system).grid(row=4, column=0, columnspan=4, sticky=W)
Button(self.__master, text="Start", command=self.__master.destroy).grid(row=5, column=0, columnspan=4)
def loop(self):
self.__master.mainloop()
self.__args["base"] = self.__base.get()
self.__args["predict_images"] = self.__predict_images.get()
self.__args["predict_images_top"] = int(self.__predict_images_top.get())
self.__args["gui_tag"] = self.__gui_tag.get()
self.__args["open_system"] = self.__open_system.get()
return self.__args
def __browse(self):
filename = filedialog.askdirectory()
self.__base.set(filename)
def __validate_number(self, P):
if str.isdigit(P) or P == "":
return True
else:
return False
class GuiTag(object):
RETURN_NEXT = 0
RETURN_ABORT = 1
def __init__(self, file, tags):
self.__ret = self.RETURN_NEXT
self.__master = Tk()
self.__tags = StringVar(self.__master, value=','.join(tags))
Label(self.__master, text="File: {}".format(file)).grid(row=0, column=0, columnspan=2)
Entry(self.__master, textvariable=self.__tags).grid(row=1, column=0, columnspan=2, sticky="we")
Button(self.__master, text="Next", command=self.__handle_next).grid(row=2, column=0)
Button(self.__master, text="Abort", command=self.__handle_abort).grid(row=2, column=1)
def loop(self):
self.__master.mainloop()
return (self.__ret, self.__tags.get().split(","))
def __handle_next(self):
self.__ret = self.RETURN_NEXT
self.__master.destroy()
def __handle_abort(self):
self.__ret = self.RETURN_ABORT
self.__master.destroy()
class GuiImage(object):
RETURN_NEXT = 0,
RETURN_ROTATE_90_COUNTERCLOCKWISE = 1,
RETURN_ROTATE_90_CLOCKWISE = 2,
RETURN_ABORT = 3
def __init__(self, img, tags):
self.__ret = self.RETURN_NEXT
self.__master = Tk()
self.__tags = StringVar(self.__master, value=','.join(tags))
self.__image_pil = Image.fromarray(img)
self.__image = ImageTk.PhotoImage(image=self.__image_pil)
self.__label = Label(self.__master, width=800, height=800, image=self.__image)
self.__label.grid(row=0, column=0, columnspan=4)
Entry(self.__master, textvariable=self.__tags).grid(row=1, column=0, columnspan=4, sticky="we")
Button(self.__master, text="↺", command=self.__handle_rotate_90_counterclockwise).grid(row=2, column=0)
Button(self.__master, text="↻", command=self.__handle_rotate_90_clockwise).grid(row=2, column=1)
Button(self.__master, text="Next", command=self.__handle_next).grid(row=2, column=2)
Button(self.__master, text="Abort", command=self.__handle_abort).grid(row=2, column=3)
def loop(self):
self.__master.mainloop()
return (self.__ret, self.__tags.get().split(","))
def __handle_rotate_90_counterclockwise(self):
self.__image_pil = self.__image_pil.rotate(90)
self.__image = ImageTk.PhotoImage(image=self.__image_pil)
self.__label.config(image=self.__image)
def __handle_rotate_90_clockwise(self):
self.__image_pil = self.__image_pil.rotate(-90)
self.__image = ImageTk.PhotoImage(image=self.__image_pil)
self.__label.config(image=self.__image)
def __handle_next(self):
self.__ret = self.RETURN_NEXT
self.__master.destroy()
def __handle_abort(self):
self.__ret = self.RETURN_ABORT
self.__master.destroy()
|
