From 507ca62a29460020b7c34e9f5dbfb27c669af95b Mon Sep 17 00:00:00 2001 From: surgan12 Date: Tue, 26 Feb 2019 17:18:38 +0530 Subject: [PATCH] to_pil_image changes --- torchvision/transforms/functional.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/transforms/functional.py b/torchvision/transforms/functional.py index c69ecb69d08..86bf361fcbb 100644 --- a/torchvision/transforms/functional.py +++ b/torchvision/transforms/functional.py @@ -117,7 +117,7 @@ def to_pil_image(pic, mode=None): elif pic.ndimension() == 2: # if 2D image, add channel dimension (CHW) - pic.unsqueeze_(0) + pic = pic.unsqueeze(0) elif isinstance(pic, np.ndarray): if pic.ndim not in {2, 3}: