From 870d2e2c3884d849d7df3473ca1ea450db4b24ff Mon Sep 17 00:00:00 2001 From: Landan Seguin Date: Wed, 19 Jan 2022 10:43:39 -0800 Subject: [PATCH] Fix padding fill with ade20k (#250) --- composer/datasets/ade20k.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/composer/datasets/ade20k.py b/composer/datasets/ade20k.py index ee3b739c42c..43fa95fda01 100644 --- a/composer/datasets/ade20k.py +++ b/composer/datasets/ade20k.py @@ -315,8 +315,7 @@ def initialize_object(self, batch_size, dataloader_hparams) -> DataSpec: r_mean, g_mean, b_mean = IMAGENET_CHANNEL_MEAN image_transforms = torch.nn.Sequential( PhotometricDistoration(brightness=32. / 255, contrast=0.5, saturation=0.5, hue=18. / 255), - PadToSize(size=(self.final_size, self.final_size), - fill=(int(r_mean * 255), int(g_mean * 255), int(b_mean * 255)))) + PadToSize(size=(self.final_size, self.final_size), fill=(int(r_mean), int(g_mean), int(b_mean)))) target_transforms = PadToSize(size=(self.final_size, self.final_size), fill=0) else: