get_image.py

import os
import glob


def get_image_paths(path):
    # Gather image paths
    image_paths = glob.iglob(os.path.join(path, "*.png"))

    # Loop through images, skip those without bounding boxes
    for image_path in image_paths:
        print(image_path)
        path, file_extension = os.path.splitext(image_path)
        bounding_box_file = path + ".txt"

        # Skip data without bounding boxes
        if not os.path.isfile(bounding_box_file):
            continue