diff --git a/data/transforms.py b/data/transforms.py index c0f02a6..1548fcc 100644 --- a/data/transforms.py +++ b/data/transforms.py @@ -414,8 +414,7 @@ def __init__(self, self.transforms = Compose([ ToTensor(), Resize(min_size=min_size, max_size=max_size), - Normalize(pixel_mean, pixel_std), - PadImage(max_size=max_size) + Normalize(pixel_mean, pixel_std) ])