aboutsummaryrefslogtreecommitdiff
path: root/util/image-to-array
blob: 72abc5c01e5733e60b6e5767cda54c5a38b3374e (plain)
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
#!/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