保存的相同数据生成不同的图像 - Python

Posted 2023-05-07

【中文标题】保存的相同数据生成不同的图像 - Python

【英文标题】：Same data saved generate different images - Python

【发布时间】：2013-11-21 23:14:40

【问题描述】：

我的代码中有两种保存图像数据的方法，一种只是将其值保存为灰度值，另一种用于生成热图图像：

def save_image(self, name):
    """
    Save an image data in PNG format
    :param name: the name of the file
    """
    graphic = Image.new("RGB", (self.width, self.height))
    putpixel = graphic.putpixel
    for x in range(self.width):
        for y in range(self.height):
            color = self.data[x][y]
            color = int(Utils.translate_range(color, self.range_min, self.range_max, 0, 255))
            putpixel((x, y), (color, color, color))
    graphic.save(name + ".png", "PNG")

def generate_heat_map_image(self, name):
    """
    Generate a heat map of the image
    :param name: the name of the file
    """
    #self.normalize_image_data()
    plt.figure()
    fig = plt.imshow(self.data, extent=[-1, 1, -1, 1])
    plt.colorbar(fig)
    plt.savefig(name+".png")
    plt.close()

代表我的数据的类是这样的：

class ImageData:
def __init__(self, width, height):
    self.width = width
    self.height = height
    self.data = []
    for i in range(width):
        self.data.append([0] * height)

为两种方法传递相同的数据

ContourMap.save_image("ImagesOutput/VariabilityOfGradients/ContourMap") ContourMap.generate_heat_map_image("ImagesOutput/VariabilityOfGradients/ContourMapHeatMap")

我将一张图片相对于另一张图片旋转。

方法一：

方法二：

我不明白为什么，但我必须解决这个问题。

任何帮助将不胜感激。 提前致谢。

【问题讨论】：

作为旁注，您为什么首先使用putpixel？这是构建图像的最慢方法，尤其是在旧的 PIL/Pillow 版本中。为什么不使用单个矢量化操作来转换数组，然后一次复制整个内容？还是使用ImageDraw？还是其他什么？

我刚刚给了你两个不同的提示，以及一个文档链接，其中包含更广泛的提示。

【参考方案1】：

显然数据是行优先格式，但您在迭代时就好像它是列优先格式一样，这会将整个数据旋转 -90 度。

快速解决方法是替换这一行：

color = self.data[x][y]

…用这个：

color = self.data[y][x]

（虽然推测data 是一个数组，所以你真的应该使用self.data[y, x]。）

更清晰的解决方法是：

for row in range(self.height):
    for col in range(self.width):
        color = self.data[row][col]
        color = int(Utils.translate_range(color, self.range_min, self.range_max, 0, 255))
        putpixel((col, row), (color, color, color))

---

这在 pyplot 文档中可能并不完全清楚，但如果您查看 imshow，它解释说它采用形状为 (n, m) 的类似数组的对象并将其显示为 MxN 图像。

【讨论】：

所以我可以像这样初始化我的数据： self.data = [] for i in range(height): self.data.append([0] * width) 来修复它？抱歉，我是 Python 新手。

@pceccon：你是想让imshow 版本旋转以匹配你的putpixel 版本，还是反过来？如果imshow 是正确的，请不要理会您的数据，并更改在save_image 中循环的方式（如我的回答）。如果ifshow 错误，请更改构建数据的方式。