調整3D圖像與5D張量tensorflow

Question

我使用3D卷積我的網絡。在我的網絡節點中，我需要將圖像從[5,50,50,10,256]調整爲[5,100,100,10,256]。我只想調整圖像的軸1和軸2的大小。

我試圖用tf.image.resize_images，但現在看來，這只是工作的3D or4D張量。

任何建議我可以做什麼？

Answer 1

沒問題，我們仍然可以使用tf.image.resize_images。我們需要做的是發送數據到tf.image.resize_images它需要的形狀是張量（4D）。

# First reorder your dimensions to place them where tf.image.resize_images needs them 
transposed = tf.transpose(yourData, [0,3,1,2,4]) 

# it is now [5,10,50,50,256] 
# but we need it to be 4 dimensions, not 5 
reshaped = tf.reshape(transposed, [5*10,50,50,256]) 

# and finally we use tf.image.resize_images 
new_size = tf.constant([ 100 , 100 ]) 
resized = tf.image.resize_images(reshaped , new_size) 

# your data is now [5*10,100,100,256] 
undo_reshape = tf.reshape(resized, [5,10,100,100,256]) 

# it is now [5,10,100,100,256] so lastly we need to reorder it 
undo_transpose = tf.transpose(undo_reshape, [0,2,3,1,4]) 

# your output is now [5,100,100,10,256]