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

image captioning completed'

Browse Source
This commit is contained in:
yunjey
2017-03-21 20:01:47 +09:00
parent ba7d5467be
commit 6f5fda14f0
7 changed files with 297 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tutorials/09 - Image Captioning/data.py
View File

@ -2,11 +2,13 @@ import torch
import torchvision.transforms as transforms
import torch.utils.data as data
import os
import sys
import pickle
import numpy as np
import nltk
from PIL import Image
from vocab import Vocabulary
sys.path.append('../../../coco/PythonAPI')
from pycocotools.coco import COCO