閉じる

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

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(),
])
Pocket

コメントを残す

あなたのメールアドレスは公開されません。必須項目には印がついています *

© 2020 CV・NLP TIPS | WordPress Theme: Annina Free by CrestaProject.