336x280(권장), 300x250(권장), 250x250, 200x200 크기의 광고 코드만 넣을 수 있습니다.
만약 image가 (28,28)인 tensor면 앞에 차원을 늘려주기위해서는
image.unsqueeze(0)
를 해주면 (1,28,28) 이 된다.
이 때 channel을 3으로 늘리고 싶으면..
image = image.repeat(3, 1, 1)
를 해주면 결과적으로 (3,28,28)인 이미지가 만들어진다.
'AI > Tip' 카테고리의 다른 글
[sklearn] train, test set 분할 (0) | 2020.02.16 |
---|---|
GPU로 학습한 체크포인트 CPU에서 적용하기 (pytorch) (2) | 2020.02.13 |
docker ssh 설치 및 sublimetext sftp & filezilla 연결하기 (0) | 2020.01.16 |
zsh: command not found: brew (0) | 2020.01.16 |
[git] 이미 push 된 파일들에 .gitignore 적용하기 (1) | 2020.01.14 |