algorithm/pytorch-tutorial
1
0
Fork 0
mirror of https://github.com/yunjey/pytorch-tutorial.git synced 2025-07-08 10:46:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #159 from Michael-Hsu/patch-1

Browse Source 
Let sample.py can generate caption for gray image.
This commit is contained in:
Yunjey Choi
2020-01-28 00:04:54 +09:00
committed by GitHub
parent 22219527c5 16691c00f8
commit 32cf716575
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tutorials/03-advanced/image_captioning/sample.py
View File

@ -14,7 +14,7 @@ from PIL import Image
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
def load_image(image_path, transform=None): def load_image(image_path, transform=None):
image = Image.open(image_path) image = Image.open(image_path).convert('RGB')
image = image.resize([224, 224], Image.LANCZOS) image = image.resize([224, 224], Image.LANCZOS)
if transform is not None: if transform is not None: