RandomErasingを挿入する位置に注意[Pytorch]

6月 13, 20207月 25, 2020

torchvision.transforms.RandomErasing の引数は torch.Tensor なので、torchvision.transforms.ToTensor によって変換しておかなければならない

正しい挿入位置

transform = transforms.Compose([
    transforms.Resize((224, 224)),
    transforms.ToTensor(),
    transforms.RandomErasing(),
])

誤った挿入位置

transform = transforms.Compose([
    transforms.RandomErasing(),
    transforms.Resize((224, 224)),
    transforms.ToTensor(),
])