如何在plot_model中删除绘图背景

from tensorflow.keras.utils import plot_model
import matplotlib.pyplot as plt

model = Sequential()
# 添加模型层
...
model.compile(...)

plot_model(model, to_file='model.png', show_shapes=True, show_layer_names=True)

image = plt.imread('model.png')
plt.imshow(image, alpha=0)  # 设置背景色为透明
plt.axis('off')  # 关闭坐标轴
plt.savefig('model_transparent.png', bbox_inches='tight', pad_inches=0)  # 保存为透明背景的图片