File size: 2,140 Bytes
d19c70c fe62bf5 d19c70c fe62bf5 d19c70c fe62bf5 d19c70c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import os
from PIL import Image
from concurrent.futures import ThreadPoolExecutor
def add_green_background_to_image(image_path, output_path, background_color=(0, 255, 0)):
"""Add a green background to an image and save it as PNG."""
with Image.open(image_path) as img:
img = img.convert("RGBA")
background = Image.new("RGBA", img.size, background_color + (255,))
combined = Image.alpha_composite(background, img)
combined.save(output_path, "PNG")
def process_image_file(input_path, output_path, background_color):
"""Process a single image file to add a green background."""
if not os.path.exists(output_path):
add_green_background_to_image(input_path, output_path, background_color)
print(f"Processed: {input_path} -> {output_path}")
else:
print(f"Skipped: {output_path} already exists")
def process_directory(input_dir, output_dir, background_color=(0, 255, 0)):
"""Recursively process a directory to add a green background to all images and convert them to PNG."""
if not os.path.exists(output_dir):
os.makedirs(output_dir)
tasks = []
with ThreadPoolExecutor() as executor:
for root, _, files in os.walk(input_dir):
for file in files:
if file.lower().endswith(('.png', '.jpg', '.jpeg')):
input_path = os.path.join(root, file)
relative_path = os.path.relpath(input_path, input_dir)
output_path = os.path.join(output_dir, os.path.splitext(relative_path)[0] + '.png')
# Ensure the output directory exists
os.makedirs(os.path.dirname(output_path), exist_ok=True)
# Submit the task to the executor
tasks.append(executor.submit(process_image_file, input_path, output_path, background_color))
# Wait for all tasks to complete
for task in tasks:
task.result()
# Example usage
input_directory = "../../background-removal-arena-v0/train/data/resized"
output_directory = "../data/resized-green/"
process_directory(input_directory, output_directory) |