[Tensorflow] (B, C, W, H)를 (B, W, H, C)로 바꾸는 방법

How to convert from (B, C, W, H) to (B, W, H, C)

 

B : Batch

C : Channel

W : width

H : Height

 

 

tensorpack 예제를 사용하다보니깐 (B, C, W, H)- channel_first 형태를 (B, W, H, C)- channel_last 형태로 바꿔야 하는 과정이 필요로했습니다. 그래서 여러 자료를 찾다가 적절한 방법을 발견했습니다. 바로 tf.transepose라는 함수를 사용하면 됩니다.

 

convert from (BCWH) to (BWHC)

layer = tf.transpose(layer, perm=[0, 2, 3, 1])

 

위에 처럼 사용하면 됩니다.

 

만약, 반대로 해야할 경우는 

 

convert from (BWHC) to (BCWH)

layer = tf.transpose(layer, perm=[0, 3, 1, 2])

 

입니다.

 

참고 자료

 

• https://github.com/tensorflow/models/issues/3167
• https://stackoverflow.com/questions/47705614/how-to-visualize-images-in-tensorboard-using-tf-summary-image-in-batch-chann