更新时间:2023-01-14 16:28:01
我最终使用了 ImageGrid 工具箱.轻松多了.
I ended up using the ImageGrid toolbox. So much easier.
from mpl_toolkits.axes_grid1 import ImageGrid
idxs = [6,7,8,9]
for idx in idxs:
fig = plt.figure(idx, (15, 10))
grid = ImageGrid(fig, 111, nrows_ncols=(3, 10), axes_pad=0.1)
for i in range(3):
grid[0].set_ylabel("Input")
grid[0].set_ylabel("Ground Truth")
grid[0].set_ylabel("Prediction")
for j in range(10):
grid[j].imshow(x_true_np[idx,j], cmap="gray")
grid[j+10].imshow(x_true_np[idx,j+10], cmap="gray")
grid[j+20].imshow(x_pred_np[idx,j], cmap="gray")