[카테고리:] Pytorch

  • [Pytorch] torch.nn.PixelShuffle/PixelUnshuffle 작동 방식

    파이토치 라이브러리 중 하나인 pixelshuffle과 pixelunsuffle 함수의 작동 방식을 간단히 살펴보려한다. Image Restoration 분야에서도 down sampling과 up sampling을 할 때 많이 활용하는 함수이다.  말 그대로 픽셀들끼리 섞어서 그룹을 만들어 채널을 줄이면서 사이즈를 늘리거나(pixel shuffle) 반대로 픽셀들끼리 그룹을 만들고 채널 별로 펼쳐서 채널을 늘리면서 사이즈를 줄이는 방식이다(pixelunsuffle). 공식 설명은 아래와 같이 되어있다. 작동 방식은 정확히 안…