def test_extract_to_list_return_np_ndarray():
image = np.zeros((300, 300, 3))
extract = Extract()
list_of_faces = extract.extract_face_to_list(image)
if type(list_of_faces[0]) is str:
assert isinstance(list_of_faces[0], str)
else:
assert isinstance(list_of_faces[0], np.ndarray)
class BigGiant:
def __init__(self, pkl_path, model_path):
self.pkl_path = pkl_path
self.model_path = model_path
self.predict = Predict(self.pkl_path, self.model_path)
self.extract = Extract()
self.mtcnn = MTCNN()
def extract_face_to_list(self, image):
list_of_faces = self.extract.extract_face_to_list(image)
return list_of_faces
def predict_face(self, face):
data = self.predict.predict_face(face)
return data
def read_image(self, image_path):
image = self.extract.read_image(image_path)
return image
def aggregate_face_data(self, data, image_url):
pass
