python_scripts/class_images.py at master · juranga/python_scripts · GitHub

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
##  Made by Gerardo Uranga
##  guranga@miners.utep.edu
##  Last Modified: 2/23/2017
##
##  Small script to place images in folders of the user's choice.
##  Runs from command line. To change, simply uncomment direc and output to
##  suit your needs.
##  Bugs tend to happen ocassionally, send me an email!

import os, sys
import shutil
import cv2

#&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&
# Import Variable Meanings:
# direc = input directory
# output = output directory
# trash = trash folder directory
# image_folder = input directory: name used so for loops are easier to understand
# folder_names[] = list of folder names that contain the names corresponding to command line inputs
# image_direcs[] = list of file names inside the folders of the image folders
# image_files[] = list of images file names
# j = used to iterate through image_direcs and in the while loop
# new_direc = updated directory of the image after being moved.
# key = button pressed on keyboard
#&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&

#direc = 'your\\directory\\here'
direc = sys.argv[1]
#output = 'your\\directory\\here'
output = sys.argv[2]
trash = output+'\\'+"trash"
if not (os.path.exists(trash)):
    os.makedirs(trash)
image_folder = os.listdir(direc)
folder_names = []
image_direcs = []
image_files = []

## method that checks images
def image_check(image_direcs):
    j =0
    while (j < len(image_direcs)):
        while(True): #fixes bug where file is a ghost file that is being read
            try:
                image = cv2.imread(image_direcs[j])
                cv2.imshow("",image)
                break
            except cv2.error as e:
                j = j+1
            except IndexError as e:
                return
        key = cv2.waitKey()
        while (key == 117 and j > 0): #undo key, exits program when trying to undo the first image choice
            j = j-1
            image = cv2.imread(image_direcs[j])
            cv2.imshow("",image)
            key = cv2.waitKey()
        if (key == 100): # delete key, sends file to a "trash" folder that gets removed once script is fully done
            new_direc = trash + '\\' + image_files[j]
            ## try sections are to prevent error when adding a file with the same name as another file to a folder
            try:
                shutil.move(image_direcs[j], trash)
            except:
                pass
        else:
            new_direc = output + '\\' + folder_names[int(chr(key))-1] + '\\' + image_files[j]
            try:
                shutil.move(image_direcs[j], output +'\\' + folder_names[int(chr(key))-1])
            except:
                pass
        image_direcs[j] = new_direc
        j += 1

for i in range(0,len(sys.argv)-3):
    folder_names.append(sys.argv[i+3])
    if not (os.path.exists(output + '\\' + folder_names[-1])):
        os.makedirs(output+'\\'+folder_names[-1])

for folders in image_folder:
    file_direc = (direc + '\\' + folders)
    for fil in os.listdir(file_direc):
        image_direcs.append(file_direc +'\\' + fil)
        image_files.append(fil)
    image_check(image_direcs)
os.rmdir(trash)