Увеличение размера изображений, отображаемых в Pytorch

Question

Я хочу отобразить несколько изображений и соответствующих им ярлыков с помощью загрузчика данных Pytorch.

Однако отображаемое изображение очень крошечная.

Как увеличить ширину каждого изображения, чтобы оно стало больше.

Вот код, который я использовал:

mean_nums = [0.485, 0.456, 0.406]
std_nums = [0.229, 0.224, 0.225]

def imshow(inp, title=None):
    """Imshow for Tensor."""
    inp = inp.numpy().transpose((1, 2, 0))
    mean = np.array(mean_nums)
    std = np.array(std_nums)
    inp = std * inp + mean
    inp = np.clip(inp, 0, 1)
    plt.imshow(inp)
    if title is not None:
        plt.title(title)
    plt.pause(0.001)  # pause a bit so that plots are updated


# Get a batch of training data
inputs, classes = next(iter(dataloaders['trainLoader']))

# Make a grid from batch
out = torchvision.utils.make_grid(inputs,nrow=2)

imshow(out, title=[image_datasets['train'].classes[x] for x in classes])

Answer 1

Попробуйте вставить plt.figure(figsize=[width, height]) перед plt.imshow и выберите width, height, который вас удовлетворит.

Так, например, функция imshow может быть:

def imshow(inp, title=None):
    """Imshow for Tensor."""
    inp = inp.numpy().transpose((1, 2, 0))
    mean = np.array(mean_nums)
    std = np.array(std_nums)
    inp = std * inp + mean
    inp = np.clip(inp, 0, 1)
    plt.figure(figsize=[20, 20])
    plt.imshow(inp)
    if title is not None:
        plt.title(title)
    plt.pause(0.001)  # pause a bit so that plots are updated