From a76c7a1839573052ea7694ea63aea31a6cd137f7 Mon Sep 17 00:00:00 2001 From: Leonard Kugis Date: Fri, 31 Mar 2023 03:16:10 +0200 Subject: Added options to choose prediction backend and skip tagged files. --- gui.py | 34 +++++++++++++++++++++------------- 1 file changed, 21 insertions(+), 13 deletions(-) (limited to 'gui.py') diff --git a/gui.py b/gui.py index f20801e..7bd30ad 100644 --- a/gui.py +++ b/gui.py @@ -1,5 +1,6 @@ from tkinter import * from tkinter import filedialog +from tkinter import ttk from PIL import Image, ImageTk import logging from enum import Enum @@ -13,6 +14,7 @@ class GuiMain(object): self.__base = StringVar(self.__master, value=args["base"]) self.__file_dir = StringVar(self.__master, value=args["file_dir"]) self.__predict_images = BooleanVar(self.__master, value=args["predict_images"]) + self.__predict_images_backend = StringVar(self.__master, value=str(args["predict_images_backend"])) self.__predict_images_top = StringVar(self.__master, value=str(args["predict_images_top"])) self.__predict_images_skip_detail = BooleanVar(self.__master, value=args["predict_images_skip_detail"]) self.__predict_images_detail_factor = StringVar(self.__master, value=str(args["predict_images_detail_factor"])) @@ -20,6 +22,7 @@ class GuiMain(object): self.__gui_image_length = StringVar(self.__master, value=str(args["gui_image_length"])) self.__open_system = BooleanVar(self.__master, value=args["open_system"]) self.__skip_prompt = BooleanVar(self.__master, value=args["skip_prompt"]) + self.__skip_tagged = BooleanVar(self.__master, value=args["skip_tagged"]) self.__index = StringVar(self.__master, value=str(args["index"])) validate_number = (self.__master.register(self.__validate_number)) @@ -31,19 +34,22 @@ class GuiMain(object): Entry(self.__master, textvariable=self.__file_dir).grid(row=1, column=1, columnspan=2) Button(self.__master, text="Browse", command=lambda: self.__browse(self.__file_dir)).grid(row=1, column=3) Checkbutton(self.__master, text="Use prediction for image tagging", variable=self.__predict_images).grid(row=2, column=0, columnspan=4, sticky=W) - Label(self.__master, text="Number of top results:").grid(row=3, column=0) - Entry(self.__master, textvariable=self.__predict_images_top, validate='all', validatecommand=(validate_number, '%P')).grid(row=3, column=1, columnspan=1) - Checkbutton(self.__master, text="Skip detail scan in image prediction", variable=self.__predict_images_skip_detail).grid(row=4, column=0, columnspan=4, sticky=W) - Label(self.__master, text="Width factor for detail scan:").grid(row=5, column=0) - Entry(self.__master, textvariable=self.__predict_images_detail_factor, validate='all', validatecommand=(validate_number, '%P')).grid(row=5, column=1, columnspan=1) - Checkbutton(self.__master, text="Show GUI for tagging", variable=self.__gui_tag).grid(row=6, column=0, columnspan=4, sticky=W) - Label(self.__master, text="Image GUI preview size:").grid(row=7, column=0) - Entry(self.__master, textvariable=self.__gui_image_length, validate='all', validatecommand=(validate_number, '%P')).grid(row=7, column=1, columnspan=1) - Checkbutton(self.__master, text="Open all files with system default", variable=self.__open_system).grid(row=8, column=0, columnspan=4, sticky=W) - Checkbutton(self.__master, text="Skip prompt for file tags", variable=self.__skip_prompt).grid(row=9, column=0, columnspan=4, sticky=W) - Label(self.__master, text="Start at index:").grid(row=10, column=0) - Entry(self.__master, textvariable=self.__index, validate='all', validatecommand=(validate_number, '%P')).grid(row=10, column=1, columnspan=1) - Button(self.__master, text="Start", command=self.__master.destroy).grid(row=11, column=0, columnspan=4) + Label(self.__master, text="Image prediction backend:").grid(row=3, column=0) + ttk.Combobox(self.__master, textvariable=self.__predict_images_backend, values=("torch", "tensorflow", "keras"), state="readonly").grid(row=3, column=1) + Label(self.__master, text="Number of top results:").grid(row=4, column=0) + Entry(self.__master, textvariable=self.__predict_images_top, validate='all', validatecommand=(validate_number, '%P')).grid(row=4, column=1, columnspan=1) + Checkbutton(self.__master, text="Skip detail scan in image prediction", variable=self.__predict_images_skip_detail).grid(row=5, column=0, columnspan=4, sticky=W) + Label(self.__master, text="Width factor for detail scan:").grid(row=6, column=0) + Entry(self.__master, textvariable=self.__predict_images_detail_factor, validate='all', validatecommand=(validate_number, '%P')).grid(row=6, column=1, columnspan=1) + Checkbutton(self.__master, text="Show GUI for tagging", variable=self.__gui_tag).grid(row=7, column=0, columnspan=4, sticky=W) + Label(self.__master, text="Image GUI preview size:").grid(row=8, column=0) + Entry(self.__master, textvariable=self.__gui_image_length, validate='all', validatecommand=(validate_number, '%P')).grid(row=8, column=1, columnspan=1) + Checkbutton(self.__master, text="Open all files with system default", variable=self.__open_system).grid(row=9, column=0, columnspan=4, sticky=W) + Checkbutton(self.__master, text="Skip prompt for file tags", variable=self.__skip_prompt).grid(row=10, column=0, columnspan=4, sticky=W) + Checkbutton(self.__master, text="Skip already tagged files", variable=self.__skip_tagged).grid(row=11, column=0, columnspan=4, sticky=W) + Label(self.__master, text="Start at index:").grid(row=12, column=0) + Entry(self.__master, textvariable=self.__index, validate='all', validatecommand=(validate_number, '%P')).grid(row=12, column=1, columnspan=1) + Button(self.__master, text="Start", command=self.__master.destroy).grid(row=13, column=0, columnspan=4) def loop(self): self.__master.mainloop() @@ -51,6 +57,7 @@ class GuiMain(object): self.__args["base"] = self.__base.get() self.__args["file_dir"] = self.__file_dir.get() self.__args["predict_images"] = self.__predict_images.get() + self.__args["predict_images_backend"] = self.__predict_images_backend.get() self.__args["predict_images_top"] = int(self.__predict_images_top.get()) self.__args["predict_images_skip_detail"] = self.__predict_images_skip_detail.get() self.__args["predict_images_detail_factor"] = int(self.__predict_images_detail_factor.get()) @@ -58,6 +65,7 @@ class GuiMain(object): self.__args["gui_image_length"] = int(self.__gui_image_length.get()) self.__args["open_system"] = self.__open_system.get() self.__args["skip_prompt"] = self.__skip_prompt.get() + self.__args["skip_tagged"] = self.__skip_tagged.get() self.__args["index"] = int(self.__index.get()) return self.__args -- cgit v1.2.1