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)

 


版权声明:本文为qq_25841513原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
THE END
< <上一篇
下一篇>>