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)