Python3 批量将npy文件转换成jpg文件,且jpg图像的颜色不会翻转
时间: 2023-06-03 17:06:12 浏览: 604
你好,关于你的问题,可以使用Python的NumPy和PIL库来实现。以下是一个简单的代码示例:
```python
import os
import numpy as np
from PIL import Image
# 设置npy文件和jpg输出目录
npy_folder = 'path/to/npy/files'
jpg_folder = 'path/to/output/folder'
# 遍历npy文件夹中的所有文件
for npy_file in os.listdir(npy_folder):
if npy_file.endswith('.npy'):
npy_path = os.path.join(npy_folder, npy_file)
# 读取npy文件
arr = np.load(npy_path)
# 将数组转换为图像并保存到jpg文件夹
img = Image.fromarray(arr.astype('uint8'), 'RGB')
jpg_path = os.path.join(jpg_folder, npy_file[:-4] + '.jpg')
img.save(jpg_path)
```
这段代码将把所有npy文件转换成RGB格式的jpg图像,并保持颜色不变。注意,如果npy文件在保存时使用了其他颜色空间(例如灰度图像),则需要相应地修改代码。希望能帮到你。
阅读全文