将单个文件夹下的npy格式批量转为jpg
时间: 2023-10-09 18:05:21 浏览: 182
Python实现小工具:批量将任意格式图片转为 JPG 格式
5星 · 资源好评率100%
您可以使用Python的Numpy和OpenCV库来实现将单个文件夹下的npy格式批量转为jpg的操作。以下是一个示例代码:
```python
import numpy as np
import cv2
import os
# 定义文件夹路径和输出路径
input_dir = 'input_folder/'
output_dir = 'output_folder/'
# 遍历所有npy文件
for file_name in os.listdir(input_dir):
if file_name.endswith('.npy'):
# 加载npy文件
npy_file = np.load(input_dir + file_name)
# 转换为图像格式
img = cv2.cvtColor(npy_file, cv2.COLOR_GRAY2BGR)
# 保存为jpg文件
cv2.imwrite(output_dir + file_name[:-4] + '.jpg', img)
```
请注意,上述代码假定所有的npy文件都是灰度图像,因此我们使用`cv2.cvtColor()`将其转换为BGR格式,然后再保存为jpg文件。如果您的npy文件保存的是彩色图像,则无需进行颜色转换。此外,`file_name[:-4]`表示去掉文件名中的后缀`.npy`,并在后面添加`.jpg`作为输出文件名。
阅读全文