pytorch对自己准备的数据集进行读取的相关文件torch代码

# -*- coding: utf-8 -*-
“““
Created on Sat Oct  6 09:19:17 2018

@author: Administrator
“““
import torch
import torchvision
from torchvision import datasets transforms
import matplotlib.pyplot as plt 
import numpy as np
import os
D=299
num=3000
change=300

# Data augmentation and normalization for training 
# Just normalization for validation
data_transforms = {
    ‘train‘: transforms.Compose（[
        transforms.RandomSizedCrop（224）
        transforms.RandomHorizontalFlip（）
        transforms.ToTensor（）
        transforms.Normalize（[0.485 0.456 0.406] [0.229 0.224 0.225]）
    ]）
    ‘val‘: transforms.Compose（[
        transforms.Scale（256）
        transforms.CenterCrop（224）
        transforms.ToTensor（）
        transforms.Normalize（[0.485 0.456 0.406] [0.229 0.224 0.225]）
    ]）
}


data_dir = ‘C:\\Users\\Administrator.SKY-20180518VHY\\Desktop\\rx‘

train_sets = datasets.ImageFolder（os.path.join（data_dir ‘train‘） data_transforms[‘train‘]）
train_loader = torch.utils.data.DataLoader（train_sets batch_size=10 shuffle=True num_workers=4）
train_size = len（train_sets