ASLImageRecognition/step04_CSVConversionPreProcess.py at main · zachcarlson/ASLImageRecognition · 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
import os
import sys
from PIL import Image, ImageOps
import numpy as np
import csv


def ConvertImagesToCSV(imageDir="", csvDir="",label=0, fromGrey=True):

    #Can't find Image Directory
    if not os.path.exists(imageDir):
        print('No image directory found.')
        return

    #Make CSV Directory if it doesn't exist
    os.makedirs(os.path.dirname(csvDir), exist_ok=True)

    #Check there is at least one image in the Image Directory
    numImg = len(os.listdir(imageDir))
    if numImg < 1:
        print('No images in cropped_frames (i.e. RBG image) directory.')
        return

    #Clear CSV file if it already exists, to avoiding appending too much data
    csv_file_path = csvDir + f"{csvDir[-2:-1]}" + "_img_pixels.csv"
    if os.path.exists(csv_file_path):
        #clear file
        open(csv_file_path, 'w').close()

    #Create CSV File from Images
    for i, infile in enumerate(os.listdir(imageDir)):
        try:
            #Open Image
            img_file = Image.open(imageDir + infile)
            #get the letter for saving
            letter = infile[0]
            value = None
            if fromGrey:
                value = np.asarray(img_file.getdata(), dtype=int).reshape((img_file.size[1], img_file.size[0]))
                value = value.flatten()
                value = np.concatenate([[label], value])
                with open(csvDir + letter + "_img_pixels.csv", 'a') as f:
                    writer = csv.writer(f)
                    writer.writerow(value)
            else:
                value = np.array(img_file)

        except IOError as e:
            print(e, "ERROR - failed to convert '%s'" % infile)


def LaunchConvert(subdirectories=True, inpath="gray_frames/", outpath="CSV_Files/", grey = True):
    print("Converting...")

    if subdirectories:
        dirs = os.listdir(inpath)
        for i,d in enumerate(dirs):
            print("Converting image files in directory:", inpath + d + '/')
            ConvertImagesToCSV(inpath + d + '/', outpath + d + '/',label=i, fromGrey = grey)
    else:
        ConvertImagesToCSV(inpath, outpath, fromGrey = grey)


args = sys.argv
if len(args) < 2:
    LaunchConvert(subdirectories=True, inpath="gray_frames/", outpath="csv_files/", grey = True)
elif len(args) < 5:
    print(
        "Not enough arguments. Please provide: 1)subdirectories (boolean), 2)inpath (relative to working directory, ending in /), 3)outpath (ending in /), 4)from greyscale images bool")
else:
    LaunchConvert(subdirectories= (args[1] == "True"),
                  inpath=args[2], outpath=args[3], grey = (args[4]=='True'))