DataLoader以及图片处理操作---------Pytorch

一、DataLoader
    torch.utils.data.DataLoader(
        # 数据集;批大小;是否乱序
        dataset,batch_size=1,shuffle=False,
        # 读取数据的进程数
        num_workers=0,
        # 当样本数不能被batchsize整除时,
        # 是否舍弃最后一批数据
        drop_last=False)
 
二、图片处理
    # 组合图片的操作
    train_transform = transforms.Compose([
        transforms.Resize((32,32)), # 缩放
        transforms.RandomCrop(32, padding=4), # 裁剪,属于数据扩增,在训练时使用、测试时不用
        transforms.ToTensor(), # 图片转为张量,同时归一化像素值从[0,255]到[0,1]
        transforms.Normalize(norm_mean,norm_std), # 标准化
    ])
    train_transform(img)

 

你可能感兴趣的:(PyTorch,深度学习基础---综合)