Function transform

data/multitask_dataset.py:121–124

()

def transform():
    return Compose([
        ToTensor(),
    ])

def BGR2RGB_toTensor(back, blur, noise, shadow, watermark, target):
    back = back[:, :, [2, 1, 0]]