Tensor to NumPy

HeyHo·2022년 10월 27일
0

  1. permute 사용
    x.permute(1,2,0).numpy()

  2. Torchvision 사용
    torchvision.transforms.ToPILImage()(x)

profile
HeyHo
Coputer vision, AI
이전 포스트

gray image가 3 channel image 로 출력될 때

다음 포스트

Image.fromarray는 np.uint8만 가능

0개의 댓글