Coder Social home page Coder Social logo

Comments (1)

8bignic8 avatar 8bignic8 commented on September 27, 2024

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 ([email protected])
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()

`

from image-sorter.

Related Issues (1)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.