Springe zum Inhalt →

Monat: Januar 2019

Jedi Tricks – Python – Bilder sortieren à la Tinder

Wenn man sein Mobiltelefon wechselt oder einfach mal nur aufräumen will, dann kann es passieren, dass meine einen Ordner mit hunderten von Bildern hat, von denen man jedoch nur dutzende Aufheben will. Typisch für so ein Fall ist der WhatsApp Medien Ordner.

Jetzt kann man natürlich mit IrfanView sich von Bild zu Bild klicken, anschauen und dann löschen. Das ist jedoch nicht schnell und löschen und weiterklicken geht nicht sehr flüssig von der Hand. Und Bilder, die man behalten will, bleiben dann immer „übrig“. Wenn man am nächsten Tag weiter sortieren will, stören sie.

Auf Android gibt es die App SlideBox, die das Problem wie bei Tinder löst. Sie zeigt ein Bild an, man swiped in eine Richtung um es zu löschen, in eine Andere um es zu behalten. Das geht locker von der Hand. Aber am PC gibt es sowas nicht.

Und da ich nicht nur immer sage, dass man mehr programmieren sollte, habe ich mich hingesetzt und quick und dirty etwas zusammengeschrieben.

Den Quelltext habe ich auf meinem Github abgelegt. Gerne zum Downloaden und verwenden. Es setzt Python, TKinter und OpenCV voraus. Python sollte meiner Meinung nach immer auf einem Rechner installiert sein. Ab der Version 3.x ist TKinter mit dabei. OpenCV bekommt man einfach per:

pip install opencv-python

Ich gehe hier kurz auf das Skript ein.

import tkinter as tk
from tkinter import filedialog
import os
import cv2
import sys
import shutil

# output folder
sort = "e:/pic_swipe/sort/"
todelete = "e:/pic_swipe/delete/"

Zuerst werden die benötigten Pakete importiert.

TKinter brauchen wir für den Dateidialog. OS, SYS und SHUTIL für das Dateihandling und CV2 ist OpenCV um die Bilder zu betrachten und auch die Cursortasten abzufragen und mit Funktionen zu versehen.

Dann werden die Ausgabeverzeichnisse definiert. Man braucht ein Ordner in der die zu Löschenden und in der die zu behaltenden Dateien abgelegt werden sollen. Ja, die Ordner habe ich im ersten Wurf hart reingecoded. Das Skript hat noch ein paar weitere Schönheitsfehler. Das sollte jedoch nicht stören, wir brauchen eine Abhilfe für unser Problem.

# File dialog for picing picture folder
root = tk.Tk()
root.withdraw()
file_path = os.path.dirname(filedialog.askopenfilename())

Diese Zeilen öffnen einen Dateidialog. Man wählt damit eine Bilddatei in dem zu sortierenden Verzeichnis aus. Das Verzeichnis wird in der Variable „file_path“ gesichert.

# content of picutre folder
picturelist = os.listdir(file_path)
index = 0

Damit wird das Verzeichnis ausgelesen. Der Inhalt in die Liste „picturelist“ gesichert und die Indexvariable initialisiert.

# create viewer windows, resize picture
cv2.namedWindow("image", cv2.WINDOW_NORMAL)
img = cv2.imread(filename = str(file_path)+"/"+str(picturelist[index]))
cv2.resizeWindow('image',800, 600)
cv2.imshow('image',img)

Mit OpenCV erstellen wir das Viewerfenster und legen es auf 800×600 Pixel fest. Und laden das erste Bild (Eintrag Index 0 der Liste). Alle Bilder, die angezeigt werden, werden auch das Format „gepresst“. Ja, das verreißt die Proportionen, aber wir wollen die Bilder nicht bearbeiten, sondern nur kurz anschauen, um dann zu entscheiden, ob ins Töpfchen oder ins Kröpfchen.

while(1):
    k = cv2.waitKeyEx(0)
    if k==27:           # wait for ESC key to exit
        cv2.destroyAllWindows()
        break

Und damit sind wir schon bei der Hauptschleife. Diese läuft mit while(1) unendlich. Mit cv2.waitKexEx(0) überprüfen wir welche Taste gedrückt wird und speichern dies in die Variable „k“.

„27“ ist der Code für die ESC Taste mit der die While Schleife unterbrochen und damit das Programm beendet wird.

    elif k==2490368:    # cursor up move to delete folder
        shutil.move(str(file_path)+"/"+str(picturelist[index]), str(todelete)+"/"+str(picturelist[index]))
        index += 1
        img = cv2.imread(filename = str(file_path)+"/"+str(picturelist[index]))
        cv2.resizeWindow('image',800, 600)
        cv2.imshow('image',img)    

„2490368“ Pfeiltaste hoch. Damit verschieben wir die aktuell angezeigte Datei in den Löschen-Ordner. Erhöhen den Index um 1 und zeigen damit das nächste Bild an. Damit entsteht der Flow. Keine weitere Taste drücken um das nächste Bild zu bearbeiten.

    elif k==2621440:    #cursor down move to sort folder
        shutil.move(str(file_path)+"/"+str(picturelist[index]), str(sort)+"/"+str(picturelist[index]))
        index += 1
        img = cv2.imread(filename = str(file_path)+"/"+str(picturelist[index]))
        cv2.resizeWindow('image',800, 600)
        cv2.imshow('image',img)

„2621440“ Pfeiltaste runter. Das Bild wird in den Sortieren-Ordner verschoben und auhc hier gleich die nächste Datei angezeigt.

    elif k==2424832:    # cursor left previous picture
        index -= 1
        if index < 0:
            index = 0
        img = cv2.imread(filename = str(file_path)+"/"+str(picturelist[index]))
        cv2.resizeWindow('image',800, 600)
        cv2.imshow('image',img)
    elif k==2555904:    # cursor right next picture
        index += 1
        img = cv2.imread(filename = str(file_path)+"/"+str(picturelist[index]))
        cv2.resizeWindow('image',800, 600)
        cv2.imshow('image',img)
    elif k==-1:         # ignore waitkey standard input
        continue
    else:               # print every other pressed key
        print(k)    

Die restlichen Abfragen sind nur noch Makulatur. „2424832“ Pfeiltaste links um zum vorherigen Bild zu gelangen und „2555904“ Pfeiltaste rechts um zum nächsten zu kommen. „-1“ brauchen wir, da waitKeyEx die Tasten immer wieder abfragt und -1 zurückgibt, wenn nichts betätigt wird. Und zum Schluss noch eine Ausgabe, wenn man eine andere Taste drückt. Mit diesem Print sieht man den Code der Taste.

Wie man unschwer erkennen kann, fange ich zwar einen Index kleiner 0 ab, aber den höchsten Index, d.h. nach der letzten Bilddatei, nicht. Auch läuft das Skript auf einen Fehler, wenn man ein Bild verschoben hat und dann zurück (Pfeil links) auf das Bild springt. Das ist dann nicht mehr da und OpenCV steigt mit einem Fehler aus.

Es gibt also noch ein paar Dinge zu erledigen. Z.B. den Code zum Bildanzeigen habe ich 5 mal „getippt“, dass schreit förmlich nach einer Subroutine.

Ihr könnt euch gerne austoben und auf mein Git die Änderungen commiten.

Bei Fragen könnt Ihr Euch gerne an mich wenden. Unten werde ich das komplette Skript nochmals einfügen, eine aktuellere Version werdet Ihr immer auf Github finden.

Viel Spaß mit dem Skript und beim Sortieren der Bilder.

import tkinter as tk
from tkinter import filedialog
import os
import cv2
import sys
import shutil

# output folder
sort = "e:/pic_swipe/sort/"
todelete = "e:/pic_swipe/delete/"

# File dialog for picing picture folder
root = tk.Tk()
root.withdraw()
file_path = os.path.dirname(filedialog.askopenfilename())

# content of picutre folder
picturelist = os.listdir(file_path)
index = 0

# create viewer windows, resize picture
cv2.namedWindow("image", cv2.WINDOW_NORMAL)
img = cv2.imread(filename = str(file_path)+"/"+str(picturelist[index]))
cv2.resizeWindow('image',800, 600)
cv2.imshow('image',img)

while(1):
    k = cv2.waitKeyEx(0)
    if k==27:           # wait for ESC key to exit
        cv2.destroyAllWindows()
        break
    elif k==2490368:    # cursor up move to delete folder
        shutil.move(str(file_path)+"/"+str(picturelist[index]), str(todelete)+"/"+str(picturelist[index]))
        index += 1
        img = cv2.imread(filename = str(file_path)+"/"+str(picturelist[index]))
        cv2.resizeWindow('image',800, 600)
        cv2.imshow('image',img)
    elif k==2621440:    #cursor down move to sort folder
        shutil.move(str(file_path)+"/"+str(picturelist[index]), str(sort)+"/"+str(picturelist[index]))
        index += 1
        img = cv2.imread(filename = str(file_path)+"/"+str(picturelist[index]))
        cv2.resizeWindow('image',800, 600)
        cv2.imshow('image',img)
    elif k==2424832:    # cursor left previous picture
        index -= 1
        if index < 0:
            index = 0
        img = cv2.imread(filename = str(file_path)+"/"+str(picturelist[index]))
        cv2.resizeWindow('image',800, 600)
        cv2.imshow('image',img)
    elif k==2555904:    # cursor right next picture
        index += 1
        img = cv2.imread(filename = str(file_path)+"/"+str(picturelist[index]))
        cv2.resizeWindow('image',800, 600)
        cv2.imshow('image',img)
    elif k==-1:         # ignore waitkey standard input
        continue
    else:               # print every other pressed key
        print(k)    
Schreibe einen Kommentar