Skip to content

Instantly share code, notes, and snippets.

@sazio
Created August 28, 2020 14:20
Show Gist options
  • Save sazio/2ebe5f735d0a4025408f7f5d4741c57e to your computer and use it in GitHub Desktop.
Save sazio/2ebe5f735d0a4025408f7f5d4741c57e to your computer and use it in GitHub Desktop.
file = fake
dump_folder = images_folder_path + "/" + file[:-4]
files = glob.glob(dump_folder + "/*")
DETECTORS = {
"cv2": detect_face_cv2,
#"mtcnn": detect_face_mtcnn
}
faces_pd = None
for key, value in DETECTORS.items():
tmp_pd = pd.DataFrame(extract_faces(files, file, detector=value), columns=["filename", "image", "boxes_" + key , "faces_" + key])
if faces_pd is None:
faces_pd = tmp_pd
else:
faces_pd = pd.merge(faces_pd, tmp_pd, on=["filename", "image"], how="left")
faces_pd.head(12)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment