pytorch 如何统一更改图片名和格式
时间: 2023-09-07 11:12:06 浏览: 150
你可以使用 Python 的 `os` 模块和 PyTorch 的 `torchvision` 模块来完成图片的重命名和格式转换。
以下是一个示例代码,将文件夹下所有的 `.jpg` 格式文件重命名为以 `image_` 开头的格式为 `.png` 的文件:
```python
import os
from PIL import Image
import torchvision.transforms as transforms
# 定义图片转换函数
transform = transforms.Compose([
transforms.Resize((256, 256)), # 图片缩放为256x256
transforms.ToTensor() # 转换为tensor
])
path = 'path/to/folder' # 文件夹路径
for file_name in os.listdir(path):
if file_name.endswith('.jpg'): # 只处理 jpg 格式文件
img_path = os.path.join(path, file_name) # 获取图片路径
img = Image.open(img_path) # 打开图片
img_tensor = transform(img) # 转换为tensor
new_file_name = f'image_{file_name[:-4]}.png' # 生成新文件名
new_img_path = os.path.join(path, new_file_name) # 生成新文件路径
Image.fromarray(img_tensor.mul(255).permute(1, 2, 0).byte().numpy()).save(new_img_path) # 保存为png格式
os.remove(img_path) # 删除原文件
```
上述代码中使用了 `PIL` 库进行图片的读取和保存,使用了 `transforms` 模块对图片进行缩放和转换为 tensor。最后通过 `os` 模块对文件进行重命名和删除操作。
阅读全文