함수 이용해 이미지 rotate 하기

TTT에서는 SSL을 활용해 이미지를 rotate시키고 classifier 가 방향을 맞추도록 해 encoder를 추가로 훈련시킨다. 이를 따로 증강을 활용하는 것이 아니라 tensor들의 위치를 바꿔준다.

def tensor_rot_90(x):
    return x.flip(2).transpose(1, 2)



plt.figure(figsize=(8, 8))

for i in range(8):
    # 원본 이미지
    plt.subplot(8, 2, 2 * i + 1)
    plt.imshow(np.transpose(images[i].numpy(), (1, 2, 0)))
    plt.axis('off')
    
    # 90도 회전 이미지
    rotated_image = tensor_rot_90(images[i])
    plt.subplot(8, 2, 2 * i + 2)
    plt.imshow(np.transpose(rotated_image.numpy(), (1, 2, 0)))
    plt.axis('off')

plt.show()

 

원본이미지와 비교했을 때 90도 회전한 이미지를 얻었다.