def save_vectors_file():
data = load_data(FLAGS.data_path)
vectorizer = Vectorizer()
logging.info('getting vectors')
img_vectors = []
genders = []
for img_path, gender_id in tqdm(data.items()):
try:
img_array = get_img(img_path)
vector = vectorizer.get_vector(img_array)
img_vectors.append(vector)
genders.append(gender_id)
except Exception as e:
logging.warning('exception: {}'.format(e))
vectorizer.close()
dim_reduction_technique = get_dim_reduction_technique(
FLAGS.dim_reduction_technique)
reduced, model = dim_reduction_technique(img_vectors, FLAGS.n_dimensions)
save_pkl_file(model, FLAGS.reducter_path)
save_pkl_file((reduced, genders), FLAGS.vectors_path)
def main():
img = get_img(FLAGS.img_path)
vectorizer = Vectorizer()
vector = vectorizer.get_vector(img)
vectorizer.close()
reducter = load_pkl_file(FLAGS.reducter_path)
reduced = reducter.transform([vector])
model = load_pkl_file(FLAGS.model_path)
output = model.predict(reduced)[0]
print('result: {}'.format(output))
