【Python编程】将格式为npy的图片转换为png格式,然后保存到另外文件夹中,并且保存的文件名与原来保持一致
将格式为npy的图片转换为png格式,然后保存到另外文件夹中,并且保存的文件名与原来保持一致
·
导入必要的库
import numpy as np
import matplotlib.pyplot as plt
import os
import scipy.misc
import imageio # 用于保存灰度图
输入输出路径变量定义
input_dir = r"D:\PApple_RGB-D-Size_dataset_v1\depthCropNpy"
output_dir = r"D:\out"
if not os.path.exists(output_dir): # 如果路径不存在,则对其进行创建
os.makedirs(output_dir)
output_dir1 = r"D:\out1"
if not os.path.exists(output_dir1):
os.makedirs(output_dir1)
遍历循环
files = os.listdir(input_dir) # 获取输入路径文件夹下的文件
for i in files:
depthmap = np.load(path1+"/" + i) # 加载读取图片
filename = str(i) # 将循环变量转换为字符串,以便后面文件名的保留命名
new_filename = os.path.splitext(filename)[0] + '.png' # 新文件名变量的定义
save_path = os.path.join(output_dir, new_filename) # 将新文件名与输出路径进行连接
plt.imsave(save_path,depthmap) # 将转换后的文件保存在输出路径下
save_path1 = os.path.join(output_dir1,new_filename)
imageio.imsave(save_path1,depthmap) # 将转换后的图片以灰度图保存在输出路径下
以上是以原文件名进行保存的情况,下面介绍以编号排序命名保存的情况。
代码如下:
import numpy as np
import matplotlib.pyplot as plt
import os
import scipy.misc
import imageio
path1 = r"D:\PApple_RGB-D-Size_dataset_v1\depthCropNpy"
files = os.listdir(path1)
n=1
for i in files:
depthmap = np.load(path1+"/" + i)
plt.imsave(r"D:\out\no " + str(n) + ".png",depthmap)
imageio.imsave(r"D:\out1\no " + str(n) + ".png",depthmap)
n = n+1
完整代码:
import numpy as np
import matplotlib.pyplot as plt
import os
import scipy.misc
import imageio # 用于保存灰度图
input_dir = r"D:\PApple_RGB-D-Size_dataset_v1\depthCropNpy"
output_dir = r"D:\out"
if not os.path.exists(output_dir): # 如果路径不存在,则对其进行创建
os.makedirs(output_dir)
output_dir1 = r"D:\out1"
if not os.path.exists(output_dir1):
os.makedirs(output_dir1)
files = os.listdir(input_dir) # 获取输入路径文件夹下的文件
for i in files:
depthmap = np.load(path1+"/" + i) # 加载读取图片
filename = str(i) # 将循环变量转换为字符串,以便后面文件名的保留命名
new_filename = os.path.splitext(filename)[0] + '.png' # 新文件名变量的定义
save_path = os.path.join(output_dir, new_filename) # 将新文件名与输出路径进行连接
plt.imsave(save_path,depthmap) # 将转换后的文件保存在输出路径下
save_path1 = os.path.join(output_dir1,new_filename)
imageio.imsave(save_path1,depthmap) # 将转换后的图片以灰度图保存在输出路径下

GitCode 天启AI是一款由 GitCode 团队打造的智能助手,基于先进的LLM(大语言模型)与多智能体 Agent 技术构建,致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话,还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力,真正做到“一句话,让 Al帮你完成复杂任务”。
更多推荐
所有评论(0)