def readImages(self, image_dir):
images = []
img_names = os.listdir(image_dir)
for name in img_names:
newImage = Image()
newImage.path = image_dir
newImage.fileName = name
images.append(newImage)
return np.array(images)
def getDataFromFolder(self, path):
files = os.listdir(path)
data = []
for file in files:
if (file[0] != '.'):
newImage = Image()
newImage.path = path
newImage.fileName = file
data.append(newImage)
return np.array(data)
def saveImage(image_dir='./today'):
img = snap()
if img == -1:
img = np.ones((1200, 1900, 3))
global img_no
img_array = np.array(img)
if image_dir!=None:
image_path = image_dir + '/image' + str(img_no) + '.png'
print('saving {}'.format(image_path))
cv2.imwrite(image_path, cv2.cvtColor(img_array, cv2.COLOR_RGB2BGR))
newImage = Image()
newImage.path = image_dir
newImage.fileName = 'image' + str(img_no) + '.png'
img_no += 1
return newImage
def readOldData(self, path, sourcePath):
with open(path, 'rb') as f:
oldData = pickle.load(f)
images, faces = oldData['images'], oldData['faces']
data, facedata = dict(), dict()
for img in images:
imageObject = Image()
imageObject.fileName = img[0]
imageObject.path = sourcePath
#imageObject.image = img[1]
data[img[0]] = imageObject
for face in faces:
if (face.img_name not in facedata):
facedata[face.img_name] = []
facedata[face.img_name].append(face)
for img_name in facedata:
data[img_name].faces = np.array(facedata[img_name])
return np.array([data[key] for key in data])
