使用Matplotlib保存图像时如何保留原始尺寸？

问题场景 #

图像增强后用Matplotlib保存，发现尺寸被修改：原图像尺寸为(1224, 370)，保存后变成(948, 284)，导致增强效果失效。相关代码如下：

import matplotlib.pyplot as plt
from PIL import Image
img = Image.open( "D:\ML_project\pro\dataset\training\images\000000.png" );
width , height = img.size
print(img.size)

plt.figure(figsize = (width /120 , height/ 120));
plt.tick_params(axis='both', which='both', bottom=False, top=False, left=False, 
right=False,labelbottom=False, labeltop=False, labelleft=False, labelright=False)

plt.axis("off")
plt.imshow(img,aspect="auto")
plt.savefig("D:\ML_project\pro\dataset2\training\images\000000.png",dpi = 120,bbox_inches = "tight",pad_inches = 0);

img =  Image.open( "D:\ML_project\pro\dataset2\training\images\000000.png" );
print(img.size)

问题原因 #

1. bbox_inches="tight"参数会自动裁剪画布周围的空白区域，但Matplotlib的布局计算逻辑会导致实际保存尺寸和预期不符；
2. 仅设置figsize和dpi，未让图像完全填充画布，存在隐藏边距被裁剪。

解决方案 #

方案1：直接用PIL保存（推荐）

如果只是保存处理后的图像，无需通过Matplotlib展示，直接使用PIL的save方法即可，不会改变图像尺寸：

from PIL import Image
# 假设img是增强后的PIL图像对象
img.save("D:\ML_project\pro\dataset2\training\images\000000.png")

方案2：调整Matplotlib参数强制匹配原图尺寸

如果必须用Matplotlib保存，修改代码确保图像完全填充画布，避免自动裁剪：

import matplotlib.pyplot as plt
from PIL import Image

img = Image.open("D:\ML_project\pro\dataset\training\images\000000.png")
width, height = img.size
print(img.size)

dpi = 120
# 创建画布，确保dpi * figsize = 原图尺寸
fig = plt.figure(figsize=(width/dpi, height/dpi), dpi=dpi)
# 添加轴并让其占满整个画布，消除边距
ax = fig.add_axes([0, 0, 1, 1])
ax.set_axis_off()
ax.imshow(img)

# 保存时禁用bbox_inches="tight"，避免自动裁剪
plt.savefig("D:\ML_project\pro\dataset2\training\images\000000.png", dpi=dpi, pad_inches=0)

img = Image.open("D:\ML_project\pro\dataset2\training\images\000000.png")
print(img.size)

内容的提问来源于stack exchange，提问作者kevin godfrey