显示数据集中的图片

weeekly

于 2021-01-28 19:16:56 发布

阅读量1.4k

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/qq_35368303/article/details/113355946

这篇博客主要介绍了如何利用PyTorch读取并显示数据集中的图片，通过参考一个具体的视频教程，展示了从chest数据集读取图片、处理和显示的过程。博主指出，只要更换数据集路径，该方法同样适用于其他数据集的图片加载和展示。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

学习目标：

学习用pytorch读取显示数据集中的图片

学习内容：

参考视频教程：https://www.bilibili.com/video/BV1xA411e7oX

视频以chest数据集为例，讲解了如何读取数据集中的图片，然后将其进行处理，最终显示出来

# 1 加载库
import torch
import torch.nn as nn
from torchvision import datasets, transforms
import os
from torch.utils.data import DataLoader
from torchvision.utils import make_grid
import numpy as np
import matplotlib.pyplot as plt


# 2 定义一个方法：显示图片(该方法是官方定义好的方法，用于显示图片，所以直接用就行了)
def image_show(inp, title=None):
    plt.figure(figsize=(10, 3))  # 设置画布大小
    inp = inp.numpy().transpose((1, 2, 0))  # 将torch.FloatTensor 转换为numpy ;transpose转置
    mean = np.array([0.485, 0.456, 0.486])  # 均值
    std = np.array([0.229, 0.224, 0.225])   # 方差
    inp = std * inp + mean
    inp = np.clip(inp, 0, 1)
    plt.imshow(in