Torch.tensor to numpy img 변환 후 이미지처리시 오류 발생.
해결법
image1copy = np.uint8(image)
'====Pytorch > Pytorch' 카테고리의 다른 글
cuda_launch_blocking=1. (0) | 2021.12.03 |
---|---|
runtimeerror: pytorchstreamreader failed reading zip archive: invalid header or archive is corrupted (0) | 2021.07.12 |
댓글