Exemplo n.º 1
0
 def __init__(self):
     self.cache_path = os.path.join(config.DATA_DIR, "MSCOCO", "cache",
                                    "cache.pkl")
     self.caption_file_names = {
         "eval": "captions_val2014.json",
         "train": "captions_train2014.json"
     }
     self.image_feature_names = {
         "eval": "eval_img_feature.pkl",
         "test": "test_img_feature.pkl",
         "train": "train_img_feature.pkl"
     }
     DataLoader.__init__(self, MSCOCODataLoader.MSCOCO_text_processor,
                         self.cache_path)
 def __init__(self):
     self.cache_path = os.path.join(Filckr8kImageProcessor.data_dir,
                                    "cache", "cache.pkl")
     DataLoader.__init__(self, Flickr8KDataLoader.Flickr8k_text_processor,
                         self.cache_path)