将文件夹里的图片缩放至统一尺寸 224 × 224的代码
时间: 2024-05-03 11:20:42 浏览: 65
将一图片同比例缩小绘制的源代码
以下是Python的代码实现:
```python
from PIL import Image
import os
def resize_images(input_dir, output_dir, new_size):
"""将文件夹里的所有图片缩放至给定的尺寸并保存到新的文件夹中。
参数:
input_dir (str): 存放原始图片的文件夹路径。
output_dir (str): 存放缩放后图片的文件夹路径。
new_size (tuple): 缩放后的尺寸,格式为 (宽度, 高度)。
"""
# 检查输出文件夹是否存在,不存在则创建。
if not os.path.exists(output_dir):
os.makedirs(output_dir)
# 遍历输入文件夹内的所有图片,并进行缩放操作。
for filename in os.listdir(input_dir):
if filename.endswith(".jpg") or filename.endswith(".png"):
image_path = os.path.join(input_dir, filename)
output_path = os.path.join(output_dir, filename)
with Image.open(image_path) as im:
im_resized = im.resize(new_size)
im_resized.save(output_path)
# 使用示例:
resize_images("input_folder", "output_folder", (224, 224))
```
其中,`input_dir` 和 `output_dir` 分别为输入图片的文件夹路径和输出缩放后图片的文件夹路径,`new_size` 为缩放后的尺寸,格式为 (宽度, 高度)。这段代码会遍历输入文件夹内的所有图片,并将它们缩放至给定的尺寸,并保存到指定的输出文件夹中。
阅读全文