从matplotlib savefig中删除空格

时间:2013-03-02 23:11:58

标签: python matplotlib

我正在尝试阅读一系列.bmp图像,并根据我得到的提示做一些线性对比度调整。这些图像很小,只有112x112,我想让它们看起来完全一样,除了对比度调整。我尝试用matplotlib做,但无论我做什么,我都会在图像的边界周围得到白色空间。这是我正在使用的代码:

# Open image and convert to array
oldImage = Image.open(f)
imageArray = np.array(oldImage)

# Preprocessing
vrange = stats.mquantiles(imageArray.flatten(),prob=[0.01,0.99])

# Plot and save
fig = plt.figure()
fig.set_size_inches(1,1)
fig.set_dpi(112)
plt.imshow(imageArray,cmap="gray",interpolation="Nearest",vmin=vrange[0],vmax=vrange[1]);
plt.axis('off')
plt.savefig(f[:-4] + "_adjusted.png", bbox_inches='tight')

有关如何移除衬垫的任何提示?我已经做了一些谷歌搜索,但到目前为止,我发现的一切都没有。

2 个答案:

答案 0 :(得分:2)

你可以在没有matplotlib的情况下进行阈值处理:

import os
from PIL import Image
import numpy as np
import scipy.stats.mstats as mstats

f = os.path.expanduser('~/tmp/image.png')
name, ext = os.path.splitext(f)
out = name+"_adjusted.png"

oldImage = Image.open(f).convert('L')
imageArray = np.array(oldImage)

vmin, vmax = mstats.mquantiles(imageArray.flatten(), prob=[0.01,0.99])

np.clip(imageArray, vmin, vmax, out=imageArray)
imageArray = (imageArray-vmin)*255/(vmax-vmin)
img = Image.fromarray(imageArray.astype('uint8'), 'L')
img.save(out)

这样,您不必以英寸为单位定义图形大小,也不必定义DPI等。您只需将PIL图像转换为numpy数组,进行一些数学运算,然后转换回PIL图像。

答案 1 :(得分:-1)

plt.savefig()之前添加以下行:

plt.subplots_adjust(0,0,1,1,0,0)