② ToTensor()(img).unsqueeze(0)에서 사용된 unsqueeze()는 1차원 데이터를 생성하는 함수입니다. 즉, 이미지 데이터를 텐서로 변환하고, 그 변환된 데이터를 1차원으로 변경하겠다는 의미입니다. unsqueeze는 다음과 같이 사용할 수 있습니다.
import torch x1 = torch.rand(3, 10, 64) x2 = x1.unsqueeze(dim=0) ------ [3, 10, 64] -> [1, 3, 10, 64] print(x2.shape) print('---------------') x3 = x1.unsqueeze(dim=1) ------ [3, 10, 64] -> [3, 1, 10, 64] print(x3.shape)
이 코드를 실행하면 다음과 같이 출력됩니다.
torch.Size([1, 3, 10, 64]) --------------- torch.Size([3, 1, 10, 64])
다음 그림은 호출한 이미지를 보여 줍니다.
▲ 그림 5-46 예제에서 사용할 이미지