2017-09-24 3 views
0

私はテンソルフローに取り組んでおり、形状が数字の配列で車の画像を持っています。(3, 512, 660, 4)(3,512,660,4)から(3,2048,660,1)へのナンキーチェンジシェイプ

3は、車のインデックスに対応し、512*660は画像サイズであり、4は車の異なる辺に対応する。

つまり、(1, 512, 660, 1)はCar1 - フロントサイドイメージ、(1, 512, 660, 2)はCar1 - 左側イメージなどに対応します。

ここでは、車のすべての画像を1つの画像(2048*660)に連結したいと考えています。つまり、(3, 512, 660, 4)から(3, 2048, 660, 1)にリサイズします。

誰かが私を助けることができますか?

私は変形機能を試しましたが、実際には画像を連結するのではなく重なっています。

答えて

2

最後の軸を新しい第3の軸として前に押して、再形成するように軸を並べ替えることができます。並べ替え軸がそうのように、私たちの3つのソリューションを与え、np.swapaxesまたはnp.transposeまたはnp.rollaxisで扱うことができ -

a.swapaxes(2,3).reshape(3,2048,660,1) 
a.transpose(0,1,3,2).reshape(3,2048,660,1) 
np.rollaxis(a,3,2).reshape(3,2048,660,1) 

あなたが前面に両面インデックスを持っていると思った場合は、それに応じて移調 -

a.transpose(0,3,1,2).reshape(3,2048,660,1) 
関連する問題