Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Hey I think your script could be even better with a requirements.txt #2

Open
8bignic8 opened this issue Jan 13, 2023 · 1 comment
Open

Comments

@8bignic8
Copy link

i know there are not much requirements but it would be good practice and also a python Version would be nice :).

@8bignic8
Copy link
Author

supports now python 3.10.7 and more than tiff if you want to integrate it thats up to you :):

`"""
Main script for sorting images in a folder into subfolders. This script launches a GUI which displays
one image after the other and lets the user vote for different given labels from a list given as input to the
script.

USAGE:

python sort_folder.py --folder --labels ...

Author: Christian Baumgartner (c.baumgartner@imperial.ac.uk)
Date: 31. Jan 2016
"""

import argparse
import tkinter as tk
import os
from shutil import copyfile, move
from PIL import ImageTk, Image

class ImageGui:
"""
GUI for iFind1 image sorting. This draws the GUI and handles all the events.
Useful, for sorting views into sub views or for removing outliers from the data.
"""

def __init__(self, master, labels, paths):
    """
    Initialise GUI
    :param master: The parent window
    :param labels: A list of labels that are associated with the images
    :param paths: A list of file paths to images
    :return:
    """

    # So we can quit the window from within the functions
    self.master = master

    # Extract the frame so we can draw stuff on it
    frame = tk.Frame(master)

    # Initialise grid
    frame.grid()

    # Start at the first file name
    self.index = 0
    self.paths = paths
    self.labels = labels

    # Number of labels and paths
    self.n_labels = len(labels)
    self.n_paths = len(paths)

    # Set empty image container
    self.image_raw = None
    self.image = None
    self.image_panel = tk.Label(frame)

    # set image container to first image
    #print(self.index)
    print(self.paths)
    self.set_image(self.paths[self.index])

    # Make buttons
    self.buttons = []
    for label in labels:
        self.buttons.append(
                tk.Button(frame, text=label, width=10, height=1, command=lambda l=label: self.vote(l))
        )

    # Add progress label
    progress_string = ("%d/%d" % (self.index, self.n_paths))
    self.progress_label = tk.Label(frame, text=progress_string, width=10)

    # Place buttons in grid
    for ll, button in enumerate(self.buttons):
        button.grid(row=0, column=ll, sticky='we')
        #frame.grid_columnconfigure(ll, weight=1)

    # Place progress label in grid
    self.progress_label.grid(row=0, column=self.n_labels, sticky='we')

    # Place the image in grid
    self.image_panel.grid(row=1, column=0, columnspan=self.n_labels+1, sticky='we')

    # key bindings (so number pad can be used as shortcut)
    for key in range(self.n_labels):
        master.bind(str(key+1), self.vote_key)

def show_next_image(self):
    """
    Displays the next image in the paths list and updates the progress display
    """
    self.index += 1
    progress_string = "%d/%d" % (self.index, self.n_paths)
    self.progress_label.configure(text=progress_string)

    if self.index < self.n_paths:
        self.set_image(self.paths[self.index])
    else:
        self.master.quit()

def set_image(self, path):
    """
    Helper function which sets a new image in the image view
    :param path: path to that image
    """
    image = self._load_image(path)
    self.image_raw = image
    self.image = ImageTk.PhotoImage(image)
    self.image_panel.configure(image=self.image)

def vote(self, label):
    """
    Processes a vote for a label: Initiates the file copying and shows the next image
    :param label: The label that the user voted for
    """
    input_path = self.paths[self.index]
    self._copy_image(input_path, label)
    self.show_next_image()

def vote_key(self, event):
    """
    Processes voting via the number key bindings.
    :param event: The event contains information about which key was pressed
    """
    pressed_key = int(event.char)
    label = self.labels[pressed_key-1]
    self.vote(label)

@staticmethod
def _load_image(path, size=(800,600)):
    """
    Loads and resizes an image from a given path using the Pillow library
    :param path: Path to image
    :param size: Size of display image
    :return: Resized image
    """
    image = Image.open(path)
    image = image.resize(size, Image.ANTIALIAS)
    return image

@staticmethod
def _copy_image(input_path, label):
    """
    Copies a file to a new label folder using the shutil library. The file will be copied into a
    subdirectory called label in the input folder.
    :param input_path: Path of the original image
    :param label: The label
    """
    root, file_name = os.path.split(input_path)
    output_path = os.path.join(root, label, file_name)
    print (" %s --> %s" % (file_name, label))
    copyfile(input_path, output_path)

@staticmethod
def _move_image(input_path, label):
    """
    Moves a file to a new label folder using the shutil library. The file will be moved into a
    subdirectory called label in the input folder. This is an alternative to _copy_image, which is not
    yet used, function would need to be replaced above.
    :param input_path: Path of the original image
    :param label: The label
    """
    root, file_name = os.path.split(input_path)
    output_path = os.path.join(root, label, file_name)
    print (" %s --> %s" % (file_name, label))
    move(input_path, output_path)

def make_folder(directory):
"""
Make folder if it doesn't already exist
:param directory: The folder destination path
"""
if not os.path.exists(directory):
os.makedirs(directory)

The main bit of the script only gets exectured if it is directly called

if name == "main":

# Make input arguments
parser = argparse.ArgumentParser()
parser.add_argument('-f', '--folder', help='Input folder where the *tif images should be', required=True)
parser.add_argument('-l', '--labels', nargs='+', help='Possible labels in the images', required=True)
args = parser.parse_args()

# grab input arguments from args structure
input_folder = args.folder
print(input_folder)
labels = args.labels

# Make folder for the new labels
for label in labels:
    make_folder(os.path.join(input_folder, label))

# Put all image file paths into a list
paths = []
all = 0
for file in os.listdir(input_folder):
    if file.endswith(".png") or file.endswith(".tiff") or  file.endswith(".jpg") or file.endswith(".tif"):
        all = all +1
        path = os.path.join(input_folder, file)
        paths.append(path)
    if (all == 0):
        print('error no pictures in folder found')

# Start the GUI
root = tk.Tk()
app = ImageGui(root, labels, paths)
root.mainloop()

`

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant