57 lines
2.1 KiB
Python
57 lines
2.1 KiB
Python
# -*- coding:utf-8 -*-
|
||
# @Author len
|
||
# @Create 2023/10/31 20:24
|
||
|
||
|
||
from PIL import Image
|
||
import os
|
||
import random
|
||
|
||
# 设置路径
|
||
background_folder = r'D:\Waste\嵌入式\数据集\背景'
|
||
overlay_image_path = r'D:\Waste\嵌入式\数据集\交通标志\禁止通行.png'
|
||
output_folder = r'D:\Waste\嵌入式\数据集\交通标志\002\禁止通行'
|
||
|
||
# 读取叠加图片
|
||
overlay = Image.open(overlay_image_path)
|
||
overlay = overlay.convert("RGBA")
|
||
|
||
# 确保输出目录存在
|
||
if not os.path.exists(output_folder):
|
||
os.makedirs(output_folder)
|
||
|
||
# 遍历背景图片文件夹中的所有文件
|
||
for background_image_name in os.listdir(background_folder):
|
||
if background_image_name.endswith(('.png', '.jpg', '.jpeg')): # 检查文件扩展名
|
||
background_path = os.path.join(background_folder, background_image_name)
|
||
background = Image.open(background_path)
|
||
background = background.convert("RGBA")
|
||
|
||
# 随机选择一个缩放因子在90%到110%之间
|
||
scale_factor = random.uniform(0.1, 0.5)
|
||
|
||
# 缩放overlay图片
|
||
new_size = (int(overlay.width * scale_factor), int(overlay.height * scale_factor))
|
||
scaled_overlay = overlay.resize(new_size, Image.Resampling.LANCZOS)
|
||
|
||
# 随机选择overlay的位置
|
||
max_x = background.width - scaled_overlay.width
|
||
max_y = background.height - scaled_overlay.height
|
||
if max_x < 0 or max_y < 0:
|
||
print(f"Scaled overlay image is larger than background {background_image_name}. Skipping.")
|
||
continue
|
||
rand_x = random.randint(0, max_x)
|
||
rand_y = random.randint(0, max_y)
|
||
|
||
# 将叠加图片放在背景图片上的随机位置
|
||
background.paste(scaled_overlay, (rand_x, rand_y), scaled_overlay)
|
||
|
||
# 将结果保存为新图片
|
||
output_path = os.path.join(output_folder, background_image_name)
|
||
# 如果输出文件格式要求为JPEG，确保转换为RGB模式
|
||
if output_path.lower().endswith('.jpg') or output_path.lower().endswith('.jpeg'):
|
||
background = background.convert('RGB')
|
||
background.save(output_path)
|
||
|
||
print("所有图片处理完成。")
|