본문 바로가기

Data handling

[numpy] 이미지 채널 축 변경

728x90

배치 사이즈가 0번째 축이라고 가정했을 때

(5, 256, 256, 3) 이라는 이미지가 있고 (B, W, H, C) 이를

(5, 3, 256, 256) 차원으로 변경하려면

np.rollaxis(3, 1) 

3번 축을 1번 축까지 굴림.