Skip to content

Commit bcc0677

Browse files
Update utils.py
1 parent 7bd7e45 commit bcc0677

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

utils/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def get_transforms(norm_mean,norm_std):
5555
width=32
5656
)
5757
], p=0.5),
58-
A.Cutout(num_holes=1, max_h_size=8, max_w_size=8, fill_value=(norm_mean[0]*255, norm_mean[1]*255, norm_mean[2]*255), p=1),
58+
A.Cutout(num_holes=1, max_h_size=16, max_w_size=16, fill_value=(norm_mean[0]*255, norm_mean[1]*255, norm_mean[2]*255), p=1),
5959
A.Rotate(limit=5),
6060
A.Normalize(norm_mean, norm_std),
6161
ToTensorV2()
@@ -89,7 +89,7 @@ def get_transforms_custom_resnet(norm_mean,norm_std):
8989
)
9090
], p=1),
9191
A.HorizontalFlip(p=1),
92-
A.Cutout(num_holes=4, max_h_size=16, max_w_size=16, fill_value=(norm_mean[0]*255, norm_mean[1]*255, norm_mean[2]*255), p=1),
92+
A.Cutout(num_holes=2, max_h_size=8, max_w_size=8, fill_value=(norm_mean[0]*255, norm_mean[1]*255, norm_mean[2]*255), p=1),
9393
A.Normalize(norm_mean, norm_std),
9494
ToTensorV2()
9595
]

0 commit comments

Comments
 (0)