#?-*-?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