matplotlib 我怎样才能正确地绘制子图?

0ve6wy6x  于 2023-01-13  发布在  其他
关注(0)|答案(1)|浏览(94)
def plot_XAI2(img, model):
  fig, axes = plt.subplots(1, 2, figsize=(12, 6))
  ax.imshow(img)
  ax.imshow(explain_image_lime(img, model))
  ax.set_title("Grad-CAM")
  ax.set_title("LIME")
  plt.show()

img = path_to_image('Lung_cancer (1).jpg')
plot_XAI2(img, model)
predict_image_class(img, model)

输出是没有任何图像的空尺寸,问题是什么?

2ic8powd

2ic8powd1#

正如@cheersmate在评论中所说的,您希望绘制到axes上,而不是ax(代码中没有定义)。axes将是一个包含两个Axes对象的列表,因此您可以改为:

def plot_XAI2(img, model):
  fig, axes = plt.subplots(1, 2, figsize=(12, 6))
  axes[0].imshow(img)  # plot in first subplot
  axes[1].imshow(explain_image_lime(img, model))  # plot in second subplot
  axes[0].set_title("Grad-CAM")
  axes[1].set_title("LIME")
  plt.show()

相关问题