同时使用yolov5、matplotlib绘图不显示问题

# Save backend
b = plt.get_backend()

torch.hub.load(yolo_path, 'custom', path=onnx_path, source='local')

#Reset backend
matplotlib.use(b)
#或者直接
import matplotlib

model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # Load model first
matplotlib.use('TkAgg') # Change backend after loading model

        if plt.get_backend()=="agg":
            plt.switch_backend("TkAgg")

import matplotlib.pyplot as plt
def plot_images(images, titles=None, ncols=2):
    """
    绘制多个图像，并自适应布局。

    参数:
        images (list): 包含图像数据的列表。
        titles (list): 图像标题列表（可选）。
        ncols (int): 每行的图像数量。
    """
    try:
        if plt.get_backend()=="agg":
            plt.switch_backend("TkAgg")

        num_images = len(images)
        print(f"本次显示{num_images}张图片")

        nrows = (num_images + ncols - 1) // ncols
        fig, axes = plt.subplots(nrows, ncols, figsize=(12, 8))

        for i, ax in enumerate(axes.flat):
            try:

                if i < num_images:

                    ax.imshow(cv2.cvtColor(images[i], cv2.COLOR_BGR2RGB))

                    ax.axis("off")
                    if titles:
                        ax.set_title(titles[i])
                else:
                    ax.axis("off")
            except Exception as e:
                print(f"{i} -- {e}")
        plt.tight_layout()
        plt.show()
    except Exception as e:
        print(e)