python一个窗口中显示多张图像(matplotlib.pyplot)

欧拉传人

于 2020-11-24 14:27:40 发布

阅读量1.4w

点赞数 16

分类专栏： python 文章标签： python

本文链接：https://blog.youkuaiyun.com/weixin_43287414/article/details/110074186

版权

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

def show_images(images, titles=None, num_cols=None, scale=3, normalize=False):
    """ 一个窗口中绘制多张图像:
    Args: 
        images: 可以为一张图像(不要放在列表中)，也可以为一个图像列表
        titles: 图像对应标题、
        num_cols: 每行最多显示多少张图像
        scale: 用于调整图窗大小
        normalize: 显示灰度图时是否进行灰度归一化
    """

    # 加了下面2行后可以显示中文标题
    plt.rcParams['font.sans-serif'] = ['SimHei']
    plt.rcParams['axes.unicode_minus'] = False

    # 单张图片显示
    if not isinstance(images, list):
        if not isinstance(scale, tuple):
            scale = (scale, scale * 1.5)

        plt.figure(figsize=(scale[1], scale[0]))
        img = images
        if len(img.shape) ==