我希望上面那个脚本支持png和jpg格式

如果您想让一个脚本支持处理 PNG 和 JPG 格式的图像，您可以使用 Python 的 Pillow 库。Pillow 是一个 Python 图像处理库，它支持许多不同的文件格式，包括 PNG 和 JPG。

首先，您需要确保已经安装了 Pillow 库。您可以在终端中使用以下命令来安装它：

pip install Pillow

然后，您可以使用以下代码来打开和保存 PNG 和 JPG 格式的图像：

from PIL import Image

# 打开图像
img = Image.open('image.png')

# 处理图像
# ...

# 保存图像为 PNG 格式
img.save('output.png')

# 保存图像为 JPG 格式
img.save('output.jpg')

在上面的代码中，`Image.open()` 函数用于打开图像文件，`img.save()` 函数用于保存图像文件。您可以根据需要在处理图像的过程中添加代码。