diff options
author | Vasil Zlatanov <v@skozl.com> | 2018-10-18 12:54:21 +0100 |
---|---|---|
committer | Vasil Zlatanov <v@skozl.com> | 2018-10-18 12:54:21 +0100 |
commit | 1afdc762b80613f9ae8c6f35296a883c40fdc375 (patch) | |
tree | 501b9b650b447719b5ea6164fe478c40877dfc70 /util | |
parent | 1bd5993aeb5cb84657cd1acb72f9b7c6e6c3553d (diff) | |
download | vz215_np1915-1afdc762b80613f9ae8c6f35296a883c40fdc375.tar.gz vz215_np1915-1afdc762b80613f9ae8c6f35296a883c40fdc375.tar.bz2 vz215_np1915-1afdc762b80613f9ae8c6f35296a883c40fdc375.zip |
Require -m argument
Diffstat (limited to 'util')
-rwxr-xr-x | util/image-to-array | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/util/image-to-array b/util/image-to-array new file mode 100755 index 0000000..72abc5c --- /dev/null +++ b/util/image-to-array @@ -0,0 +1,35 @@ +#!/usr/bin/env python +import numpy as np +import sys, os + +from PIL import Image; + +def jpg_image_to_array(image_path): + """ + Loads JPEG image into 3D Numpy array of shape + (width, height, channels) + """ + with Image.open(image_path) as image: + im_arr = np.frombuffer(image.tobytes(), dtype=np.uint8) + # im_arr = im_arr.reshape((image.size[1], image.size[0], 3)) + return im_arr + +def main(files): + for i in range(0, len(files)): + inputfile = files[i] + print(inputfile); + + if not os.path.exists(inputfile): + sys.exit('ERROR: File %s was not found!' % inputfile) + + print(jpg_image_to_array(inputfile)) + +if __name__ == "__main__": + if len(sys.argv) > 1: + main(sys.argv[1:]) + else: + print("%s: You must specify files to convert to vectors" % sys.argv[0]) + +# def compute_average_vector + +# def compute_covariance_of_normalised_faces |