added kaggle comp solution for facial keypoint

This commit is contained in:
Aladdin Persson
2021-06-05 13:26:29 +02:00
parent 8136ee169f
commit c19bb8675e
7 changed files with 27449 additions and 0 deletions

View File

@@ -0,0 +1,19 @@
import numpy as np
import pandas as pd
import os
from PIL import Image
def extract_images_from_csv(csv, column, save_folder, resize=(96, 96)):
if not os.path.exists(save_folder):
os.makedirs(save_folder)
for idx, image in enumerate(csv[column]):
image = np.array(image.split()).astype(np.uint8)
image = image.reshape(resize[0], resize[1])
img = Image.fromarray(image, 'L')
img.save(save_folder+f"img_{idx}.png")
csv = pd.read_csv("test.csv")
extract_images_from_csv(csv, "Image", "data/test/")