transforms_ft = Pipeline([ load_img_msk, # <-- New load func for img and mask RandomResizedCrop(200), # Applied to both img and mask ToTensor(), # Applied to both img and mask IntToFloatTensor(), # Only applied to img Normalize.from_stats(*imagenet_stats,cuda=False) # Only applied to img ]) out = transforms_ft(fn) out show_images((out[0][0], out[1]))