-
파이토치 torchvision.utils.save_image파이토치&텐서플로우&관련코딩 2021. 4. 22. 23:42
torchvision.utils.save_image(tensor, file_path, normalize=False, nrow=4)
만약 텐서가 0에서 1사이값이면 normalize는 해도되고 안해도되고, 텐서가 0에서 255면 해야된다
tensor가 cuda에 있는 cpu든 상관없다
'파이토치&텐서플로우&관련코딩' 카테고리의 다른 글
LR Schedulr 정리 (0) 2021.04.25 view vs reshape (0) 2021.04.23 모델 저장 (0) 2021.04.22 cv2.gaussianblur 계산하는법 (0) 2021.04.17 BCE & BCEwithLOGIT & CE (0) 2021.02.15