Skip to content

GitLab

"examples/git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "eb8e8dc84fa6b3e3174bfdc82b19035a624f7c3d"
Commit 55b5a6b8 authored by Surgan Jandial's avatar Surgan Jandial Committed by Francisco Massa
Browse files

to_pil_image changes (#762)

parent 22bc44ed
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/transforms/functional.py
View file @ 55b5a6b8
...@@ -117,7 +117,7 @@ def to_pil_image(pic, mode=None): ...@@ -117,7 +117,7 @@ def to_pil_image(pic, mode=None):
elif pic.ndimension() == 2: elif pic.ndimension() == 2:
# if 2D image, add channel dimension (CHW) # if 2D image, add channel dimension (CHW)
pic.unsqueeze_(0) pic = pic.unsqueeze(0)
elif isinstance(pic, np.ndarray): elif isinstance(pic, np.ndarray):
if pic.ndim not in {2, 3}: if pic.ndim not in {2, 3}:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment