Python 图像比较：并排显示两个图像数组

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()

def show_compare(a, b):
    fig, axs = plt.subplots(1, 2, figsize=(10, 5))
    
    if a.ndim == 2:
        axs[0].imshow(a, cmap='gray')
    else:
        axs[0].imshow(cv.cvtColor(a, cv.COLOR_BGR2RGB))
    
    if b.ndim == 2:
        axs[1].imshow(b, cmap='gray')
    else:
        axs[1].imshow(cv.cvtColor(b, cv.COLOR_BGR2RGB))
    
    plt.show()