使用pytorch读取文件夹"C:\\Users\\Admin\\Desktop\\在职研\\mission\\图像处理\\neibu-US"中所有图像,并将其变为224×224大小,取代原文件
时间: 2024-03-12 20:42:09 浏览: 140
PyTorch基础:使用卷积神经网络进行CIFAR-10图像分类(包含详细的完整的程序和数据)
使用pytorch读取文件夹中的所有图像,并将其变为指定大小的操作可以通过以下步骤实现[^1]:
1. 导入必要的库和模块:
```python
import os
from PIL import Image
import torchvision.transforms as transforms
```
2. 定义图像处理函数:
```python
def process_image(image_path, output_path, size):
# 打开图像
image = Image.open(image_path)
# 转换图像大小
transform = transforms.Resize(size)
resized_image = transform(image)
# 保存图像
resized_image.save(output_path)
```
3. 遍历文件夹中的所有图像文件,并进行处理:
```python
# 文件夹路径
folder_path = "C:\\Users\\Admin\\Desktop\\在职研\\mission\\图像处理\\neibu-US"
# 输出文件夹路径
output_folder_path = "C:\\Users\\Admin\\Desktop\\在职研\\mission\\图像处理\\neibu-US-resized"
# 确保输出文件夹存在
if not os.path.exists(output_folder_path):
os.makedirs(output_folder_path)
# 遍历文件夹中的所有图像文件
for filename in os.listdir(folder_path):
if filename.endswith(".jpg") or filename.endswith(".png"):
# 图像路径
image_path = os.path.join(folder_path, filename)
# 输出图像路径
output_path = os.path.join(output_folder_path, filename)
# 处理图像
process_image(image_path, output_path, (224, 224))
```
这样,文件夹中的所有图像都会被读取并转换为指定大小的图像,并保存在指定的输出文件夹中。
阅读全文