search

Pipe-Runner-Lab / CVPR2020-FGVC7

CVPR2020-FGVC7 Kaggle submission
MIT License
0 stars 0 forks source link

Sample code #13

Open Pipe-Runner opened 4 years ago

Pipe-Runner commented 4 years ago 
self.train_transform = Compose([HorizontalFlip(p=0.5),
                                  VerticalFlip(p=0.5),
                                  ShiftScaleRotate(rotate_limit=25.0, p=0.7),
                                  OneOf([IAAEmboss(p=1),
                                         IAASharpen(p=1),
                                         Blur(p=1)], p=0.5),
                                  IAAPiecewiseAffine(p=0.5)])
        self.test_transform = Compose([HorizontalFlip(p=0.5),
                                       VerticalFlip(p=0.5),
                                       ShiftScaleRotate(rotate_limit=25.0, p=0.7)])
        self.default_transform = Compose([Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225), always_apply=True),
                                         ToTensor()]) #normalized for pretrained network
Pipe-Runner commented 4 years ago

Detach variables to save memory

Pipe-Runner commented 4 years ago

Confusion matrix

Pipe-Runner commented 4 years ago

avg two runs

Pipe-Runner commented 4 years ago

lr: 8x10^-4 decay: 10^-3