import numpy as np import matplotlib.pyplot as plt import cv2 as cv

def show(img): if img.ndim == 2: plt.imshow(img, cmap='gray') else: plt.imshow(cv.cvtColor(img, cv.COLOR_BGR2RGB)) plt.show()

将两个图像并排显示

fig, axes = plt.subplots(1, 2, figsize=(10, 5)) # 创建一个包含两个子图的图

axes[0].imshow(a, cmap='gray' if a.ndim == 2 else None) # 显示第一个图像 axes[0].set_title('图像 A')

axes[1].imshow(b, cmap='gray' if b.ndim == 2 else None) # 显示第二个图像 axes[1].set_title('图像 B')

plt.tight_layout() # 调整布局，避免子图重叠 plt.show()