PyTorch에서는 이미지를 [channels, height, width]로 표현하는 반면 matplotlib에서는 [height, width, channels]로 표현합니다.
따라서 채널 순서를 바꿔주는 permute함수를 활용해 matplotlib에서 기대하는 채널 순서로 바꿔줍니다.
저는 서영준 입니다.
PyTorch에서는 이미지를 [channels, height, width]로 표현하는 반면 matplotlib에서는 [height, width, channels]로 표현합니다.
따라서 채널 순서를 바꿔주는 permute함수를 활용해 matplotlib에서 기대하는 채널 순서로 바꿔줍니다.