PyTorch & ファインチューニングする時に入力画像のサイズを変更する方法

Bull

画像分類のアプリ作成でCenterCropだとうまく行かなかったので（当然）Resize を利用。その時にハマった内容を備忘録として残します。

# 前処理
transform = transforms.Compose([
       transforms.Resize((224, 224)),
       transforms.ToTensor(),
       transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
])

最初、以下のように記述していましたがこれは誤り。size = 224, interpolation=224となってしまうので。

transforms.Resize(224, 224),

size = 224 x 224 とするためには

transforms.Resize((224, 224)),

のように(224, 224) として記述する必要あり。


この記事が気に入ったらサポートをしてみませんか？