-
Notifications
You must be signed in to change notification settings - Fork 0
/
transform.py
24 lines (20 loc) · 985 Bytes
/
transform.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import torchvision.transforms as transforms
from randomerasing import RandomErasing
def Train_Transform(RandomER=False):
train_transform_list = [transforms.Resize((256, 128)),
transforms.RandomHorizontalFlip(0.5),
transforms.Pad(10),
transforms.RandomCrop((256, 128)),
transforms.ToTensor(),
transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])]
if RandomER:
REA = RandomErasing()
train_transform_list.append(REA)
train_transform = transforms.Compose(train_transform_list)
return train_transform
def Val_Transform():
val_tansform_list = [transforms.Resize((256, 128)),
transforms.ToTensor(),
transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])]
val_transorm = transforms.Compose(val_tansform_list)
return val_transorm