Update utils.py

RohinSequeira · web-flow · commit 7bd7e45f0c8a · 2021-07-15T21:42:13.000+05:30
diff --git a/utils/utils.py b/utils/utils.py
@@ -55,7 +55,7 @@ def get_transforms(norm_mean,norm_std):
                 width=32
             )
         ], p=0.5),
-        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),
+        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),
         A.Rotate(limit=5),
         A.Normalize(norm_mean, norm_std),
         ToTensorV2()

Original file line number	Diff line number	Diff line change
`@@ -55,7 +55,7 @@ def get_transforms(norm_mean,norm_std):`
`55`	`55`	`width=32`
`56`	`56`	`)`
`57`	`57`	`], p=0.5),`
`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),`
	`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),`
`59`	`59`	`A.Rotate(limit=5),`
`60`	`60`	`A.Normalize(norm_mean, norm_std),`
`61`	`61`	`ToTensorV2()`