Fix padding fill with ade20k (mosaicml#250)

Landanjs · Feb 14, 2022 · 4d29e6f · 4d29e6f
1 parent 1c769ee
commit 4d29e6f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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: