import os def load_images_from_json(line_segments): line_images = [] image_paths = [] for key, value in line_segments.items(): line_images.append(value["image"]) image_paths.append(value.get("image_path", f"{key}.png")) return line_images, image_paths