使用matplotlib/pytorch从批处理中创建张量的图像网格

Question

我正在尝试从一批张量的图像网格（例如3 x 3）创建，这些张量将在下一步中通过数据加载器馈入GAN。使用以下代码，我能够将张量转换为以正确位置的网格显示的图像。问题是，它们全部显示在单独的网格中，如下所示：图1 a href =“ https://i.sstatic.net/tcjww.png” rel =“ nofollow noreferrer”>图2 图5 。如何将它们放在一个网格中，然后将一个数字带有所有9张图像？也许我使它变得太复杂了。 ：d最终必须将the the the示例张开并放入网格中。

real_samples = next(iter(train_loader))
for i in range(9):
    plt.figure(figsize=(9, 9))
    plt.subplot(330 + i + 1)
    plt.imshow(np.transpose(vutils.make_grid(real_samples[i].to(device)
                      [:40], padding=1, normalize=True).cpu(),(1,2,0)))
    plt.show()

Answer 1

这是如何使用matplotlib XD显示可变数量的奇妙加密库，

import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import ImageGrid
import matplotlib.image as mpimg

row_count = 3
col_count = 3

cryptopunks = [
    mpimg.imread(f"cryptopunks/{i}.png") for i in range(row_count * col_count)
]

fig = plt.figure(figsize=(8., 8.))
grid = ImageGrid(fig, 111, nrows_ncols=(row_count, col_count), axes_pad=0.1)

for ax, im in zip(grid, cryptopunks):
    ax.imshow(im)

plt.show()

/I.SSTATIC.NET/7X7GI.PNG“ ALT =”在此处输入图像描述>

请注意，代码允许您生成所需的所有图像，而不仅仅是3次3。我有一个文件夹cryptopunks带有许多称为＃的图像。PNG（例如，1.png，...，34.png 代码>，...）。只需更改row_count和col_count变量值。例如，对于row_count = 6和col_count = 8您得到：


如果您的图像文件没有上面的命名模式（即，只是随机名称），只需将第一行替换为以下一行：（

import os
from pathlib import Path

import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import ImageGrid
import matplotlib.image as mpimg

for root, _, filenames in os.walk("cryptopunks/"):
    cryptopunks = [
        mpimg.imread(Path(root, filename)) for filename in filenames
    ]

row_count = 3
col_count = 3

# Here same code as above.

我已经从 kaggle 。

Answer 2

对于9个图像网格，您可能需要三行和三列。最简单的方法可能就是这样：

import matplotlib.pyplot as plt

fig, axes = plt.subplots(nrows=3, ncols=3)
axes = axes.flatten() # Flattens the array so you can access individual axes

for ax in axes:
    # Do stuff with your individual axes here
    plt.show() # This call is here just for example, prob. better call outside of the loop

输出以下轴配置

它以plt.tight_layout（）：

“ rel =“ nofollow noreferrer”> matplotlib mosaic mosaic mosaic 功能功能/stable/_as_gen/matplotlib.gridspec.gridspec.html“ rel =” nofollow noreferrer“> gridspec One 。希望这会有所帮助。

编辑：这是一个解决方案，可以注释每个图的编号，以便您可以看到什么也是如此：

import matplotlib.pyplot as plt

fig, axes = plt.subplots(nrows=3, ncols=3)
axes = axes.flatten()

for idx, ax in enumerate(axes):
    ax.annotate(f"{idx+1}", xy=(0.5, 0.5), xytext=(0.5, 0.5))

Answer 3

Matplotlib提供了一个称为子图的函数，我认为这就是您要搜索的！

我猜是语法。

然后配置您的图