matplotlib保存伪彩色图像，去除白框、白边、坐标轴

去除白边

# matplotlib默认0.1的白边，将pad_inches设为-0.1
plt.savefig('name.jpg', bbox_inches='tight', dpi=100, pad_inches=-0.1)

去除坐标轴

ax.set_xticks([])
ax.set_yticks([])
ax.axis('off')

整体代码

import os.path
import glob
import numpy as np
from PIL import Image
import matplotlib.pyplot as plt

def save_img(pngfile, outdir)
    im_depth = Image.open(pngfile)
    gray_array = np.array(im_depth)
    img_s = gray_array[:, :, 0]  # 只取三通道灰度图像最后一个通道

    fig, ax = plt.subplots()
    ax.imshow(img_s, cmap='viridis')  # 转为伪彩色图像
    # 去除坐标轴
    ax.set_xticks([])
    ax.set_yticks([])
    ax.axis('off')
    if not os.path.exists(outdir):
        os.makedirs(outdir)
    # 保存图像
    plt.savefig(os.path.join(outdir, os.path.basename(pngfile)), bbox_inches='tight', dpi=100, pad_inches=-0.1)
    plt.close()

path = 'file_path'
# 迭代整个文件夹下所有的图像
img_names = glob.glob(os.path.join(path, "*"))  # 输入文件夹 output
num_images = len(img_names)

for ind, img_name in enumerate(img_names):
    print("  processing {} ({}/{})".format(img_name, ind + 1, num_images))
    save_img(img_name, "results")  # 输出文件夹 
print("finished")