In [ ]:

    

%matplotlib inline

from pycocotools.coco import COCO
import os
import numpy as np
import skimage.io as io
import matplotlib.pyplot as plt
import pickle
from nltk.corpus import wordnet as wn
plt.rcParams['figure.figsize'] = (10.0, 8.0)
from CaptionSaliency import CaptionSaliency as CS


    

In [ ]:

    

dataType='val2014'
usingSet='5000coco'
dataDir='H:/SG_code/Dataset/COCO/tools'   #<====coco path
savefileDir = 'data'

CapSal_train = CS(dataType,usingSet,dataDir,savefileDir)


    

In [ ]:

    

CapSal_train.compute_distance()