根据已有的几百张图像生成大量图像数据并保存的python代码
时间: 2024-02-18 20:05:01 浏览: 118
根据已有的几百张图像生成大量图像数据并保存可以使用 Python 中的 Pillow 库和 os 库来实现。以下是一个简单的代码示例:
```python
from PIL import Image
import os
input_dir = 'input' # 输入文件夹
output_dir = 'output' # 输出文件夹
num_images = 1000 # 需要生成的图像数量
if not os.path.exists(output_dir):
os.makedirs(output_dir)
# 遍历输入文件夹下的所有图像文件
for filename in os.listdir(input_dir):
if filename.endswith('.png'):
# 打开图像文件
img = Image.open(os.path.join(input_dir, filename))
# 获取图像的宽度和高度
width, height = img.size
# 生成指定数量的新图像
for i in range(num_images):
# 创建一个新图像
new_img = Image.new('RGB', (width, height), 'white')
# 从输入图像复制像素数据
new_img.putdata(list(img.getdata()))
# 保存新图像到输出文件夹
new_filename = os.path.splitext(filename)[0] + f'_{i+1}.png'
new_img.save(os.path.join(output_dir, new_filename))
print(f'Saved {new_filename}')
print('Done')
```
这段代码假设您已经将所有的输入图像文件保存在一个名为 input 的文件夹中,并且需要生成指定数量的新图像。它会遍历输入文件夹下的所有图像文件,然后生成指定数量的新图像并将它们保存到一个名为 output 的文件夹中。新图像的文件名将以原始文件名和一个序号的组合形式命名。例如,如果输入文件名为 example.png,那么生成的第一个新图像的文件名将为 example_1.png。您可以根据自己的需求修改代码来处理不同的图像数据。
阅读全文