导入必要的库

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)   # 将转换后的图片以灰度图保存在输出路径下
Logo

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

更多推荐