mirror of
https://github.com/deepfakes/faceswap
synced 2025-06-07 19:05:02 -04:00
28 lines
838 B
Python
28 lines
838 B
Python
import cv2
|
|
import numpy
|
|
|
|
from pathlib import Path
|
|
from scandir import scandir
|
|
|
|
image_extensions = [".jpg", ".jpeg", ".png"]
|
|
|
|
def get_folder(path):
|
|
output_dir = Path(path)
|
|
# output_dir.mkdir(parents=True, exist_ok=True)
|
|
return output_dir
|
|
|
|
|
|
def get_image_paths(directory):
|
|
return [x.path for x in scandir(directory) if
|
|
any(map(lambda ext: x.name.lower().endswith(ext), image_extensions))]
|
|
|
|
|
|
def load_images(image_paths, convert=None):
|
|
iter_all_images = (cv2.imread(fn) for fn in image_paths)
|
|
if convert:
|
|
iter_all_images = (convert(img) for img in iter_all_images)
|
|
for i, image in enumerate(iter_all_images):
|
|
if i == 0:
|
|
all_images = numpy.empty((len(image_paths), ) + image.shape, dtype=image.dtype)
|
|
all_images[i] = image
|
|
return all_images
|