caption-gen / utils /helpers.py
Sher1988's picture
Update utils/helpers.py
8e3f294 verified
raw
history blame contribute delete
202 Bytes
# from enum import Enum
import os
IMAGE_DIR = 'flickr8k/images'
DATA_DIR = 'data'
VOCAB_PATH = os.path.join(DATA_DIR, 'vocabulary.json')
CAPTIONS_PATH = os.path.join(DATA_DIR, 'captions_tokenized.csv')