torchvision.utils.make_grid의 조인 순서 확인

1971 단어 PyTorchtorchvision

이게 뭐야


  • torchvision.utils.make_grid로 결합 된 텐서가 가로 또는 세로 방향으로 결합되는지 확인합니다.
  • 결론 : Z 글자의 방향으로 결합한다.

  • 실행할 코드


  • save_image는 make_grid를 호출하는 사양이므로 save_image에 직접 tensor를 전달합니다.
  • 조금씩 어두워지는 타일을 생성해, 순서대로 늘어놓는다
  • import torch
    from torchvision.utils import save_image
    imgs = list()
    for i in range(9):
        imgs.append(torch.ones([1, 64, 64])*i/10)
    imgs = torch.stack(imgs)
    save_image(imgs, "grid.png", 3)
    

    출력





    결론


  • 왼쪽 상단에서 행 방향으로 진행
  • Z자의 방향으로 결합한다.
  • 좋은 웹페이지 즐겨찾기