批量将numpy格式文件夹转换成图片格式并保存到新的文件夹
时间: 2023-12-20 22:06:24 浏览: 183
好的,理解了。关于你的问题,你可以借助opencv库的imwrite函数进行操作,可以通过以下代码实现:
```python
import os
import numpy as np
import cv2
# 旧文件夹路径
old_folder_path = 'path/to/numpy/files/'
# 新文件夹路径
new_folder_path = 'path/to/save/images/'
# 获取旧文件夹下的所有numpy文件
files = os.listdir(old_folder_path)
# 遍历文件列表,读取numpy文件并保存为图片
for file in files:
# 获取numpy文件的绝对路径
file_path = os.path.join(old_folder_path, file)
# 读取numpy文件为numpy数组
np_array = np.load(file_path)
# 将numpy数组转换为BGR格式的图像
img = cv2.cvtColor(np_array, cv2.COLOR_RGB2BGR)
# 生成新文件的路径
new_file_path = os.path.join(new_folder_path, '{}.jpg'.format(os.path.splitext(file)[0]))
# 将图像保存为JPEG格式的文件
cv2.imwrite(new_file_path, img)
print('File {} saved successfully.'.format(file))
```
请将上面的代码中的 `path/to/numpy/files/` 改为你实际存放numpy文件的文件夹路径,将 `path/to/save/images/` 改为你实际想要保存图片的文件夹路径,然后将其保存为一个 Python 文件并运行即可。执行完毕后,你应该可以在新文件夹里找到保存的图像文件了。
阅读全文