matplotlib 使用 plt.savefig() 输出图片去除旁边的空白区域

最近在作图时需要将输出的图片紧密排布，还要去掉坐标轴，同时设置输出图片大小。

但是发现matplotlib使用plt.savefig()保存的图片

周围有一圈空白。那么如何去掉该空白呢？

首先，关闭坐标轴显示：

plt.axis('off')

但是，这样只是关闭显示而已，透明的坐标轴仍然会占据左下角位置，导致输出的图片偏右。

要想完全去掉坐标轴，需要改为以下代码：

plt.axis('off')

fig = plt.gcf()

fig.set_size_inches(7.0/3,7.0/3) #dpi = 300, output = 700*700 pixels

plt.gca().xaxis.set_major_locator(plt.NullLocator())

plt.gca().yaxis.set_major_locator(plt.NullLocator())

plt.subplots_adjust(top = 1, bottom = 0, right = 1, left = 0, hspace = 0, wspace = 0)

plt.margins(0,0)

fig.savefig(out_png_path, format='png', transparent=True, dpi=300, pad_inches = 0)

即可完成去掉空白。

注：如果不采用 subplot_adjust + margin(0,0)，而是在fig.savefig()的参数中添加bbox_inches = 'tight'，也可以达到

去除空白的效果； 但是，这样会导致对图片输出大小的设置失效。

将得到的灰度图全部转换成伪彩色显示并保存的代码

import matplotlib.pyplot as plt
import cv2
import numpy as np
import os
from os.path import join

path = r'./result2015'
file_list = './lists/kitti2015_train.list'
color_dir = './color2015'
if not os.path.exists(color_dir):
    os.mkdir(color_dir)

f = open(file_list, 'r')
filelist = f.readlines()[:10]
fig = plt.gcf()
fig.set_size_inches(12.42/3, 3.75/3)

for index in range(len(filelist)):
    current_file = filelist[index]
    imageName = current_file[0: len(current_file) - 1]
    imagePath = join(path, imageName) 
    image = cv2.imread(imagePath, cv2.COLOR_BGR2GRAY)
    
    plt.subplots_adjust(top = 1, bottom = 0, right = 1, left = 0, hspace = 0, wspace = 0)
    plt.margins(0,0)
    sc = plt.imshow(image)
    sc.set_cmap('jet')
    plt.axis('off')
    fig.savefig(join(color_dir, imageName), format='png', transparent=True, dpi=300, pad_inches = 0)

plt.show()
           

转载: https://blog.csdn.net/jifaley/article/details/79687000