web-dev-qa-db-ja.com

Pytorchテンソルのサイズを変更する方法

Pytorchのサイズtorch.Tensorの(5, 1, 44, 44)を持っています。

  • 5 =バッチサイズ
  • 1 =チャンネル
  • 44 =画像の高さ
  • 44 =画像幅

そして私はそれを形作るように(5, 1, 224, 224)

どうやってやるの？どの機能を使うべきですか？

computer-visionpytorchtensor
6
Gerwe1s_Ji

interpolate （nn.functionalの関数）を探しています。

import torch.nn.functional as nnf

x = torch.Rand(5, 1, 44, 44)
out = nnf.interpolate(x, size=(224, 224), mode='bicubic', align_corners=False)

あなたが補間の精度を本当に気にかけているならば、あなたは ResizeRight ：あらゆる種類を正確に扱うPytorch/numpyパッケージを持っているべきです "画像のサイズ変更時のエッジケース。これは、異なるスケールの特徴を直接マージするときに効果があります。

11
Shai