-
Notifications
You must be signed in to change notification settings - Fork 0
/
imageconverter.py
41 lines (26 loc) · 1.04 KB
/
imageconverter.py
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
from PIL import Image
import numpy as np
class ImageConverter(object):
def resizeImage(self, image):
return image.resize((28,28), Image.LANCZOS)
def imageToNPArray(self, path):
im = Image.open(path)
if im.size != (28,28):
im = self.resizeImage(im)
rPix = np.array(list(im.getdata(band=0)))
gPix = np.array(list(im.getdata(band=1)))
bPix = np.array(list(im.getdata(band=2)))
#Convert to one-color grayscale
pixels = np.sqrt((rPix**2 + gPix**2 + bPix**2)/3)
return pixels
def invertImage(self, pixArray):
return 255 - pixArray
def convToBlackWhiteImage(self, pixArray):
#pixArray should be 1 dimensional
bwArray = [(255 if p > 127 else 0) for p in pixArray]
return np.array(bwArray)
def loadImageAsArray(self, path):
pixArray = self.imageToNPArray(path)
pixArray = self.invertImage(pixArray)
pixArray = self.convToBlackWhiteImage(pixArray)
return pixArray/255