import os import random from PIL import Image __all__ = ["background_image_generator"] def background_image_generator(path="./dataset/pixivimages"): image_list = os.listdir(path) image_list = [os.path.join(path, image) for image in image_list] while True: yield random.choice(image_list)