# Provide paths to your test images import os import random def get_image_names(): # Get background images background_dir = "public/images/backgrounds" background_images = [f for f in os.listdir(background_dir) if f.endswith(('.jpg', '.jpeg', '.png'))] # Get people images people_dir = "public/images/people" people_images = [f for f in os.listdir(people_dir) if f.endswith(('.jpg', '.jpeg', '.png'))] generate_testing_list = [] random.shuffle(background_images) random.shuffle(people_images) total_length = len(people_images) # Get 10 random pairs for background in background_images: person = people_images[random.randint(0, total_length - 1)] generate_testing_list.append(list(( os.path.join(people_dir, person), os.path.join(background_dir, background), 10, 100))) return generate_testing_list