def __init__(self, batch_size, subset='kptest', feat_type='res5c', version='v1', use_fb_data=False): self._batch_size = batch_size self._num_top_ans = 2000 self._subset = subset self._images = None self._quest_len = None self._quest_ids = None self._quest = None self._mask = None self._answer = None self._num = None self._valid_ids = None self._vqa_image_ids = None self.use_fb_data = use_fb_data self._version_suffix = 'v2_' if version == 'v2' else '' self._load_data() self._feat_type = feat_type.lower() self._FEAT_ROOT = get_feature_root(self._subset, self._feat_type) self._transpose_feat = self._feat_type == 'res5c' # self._idx = 0 self._index = np.arange(self._num) self._is_test = True
def __init__(self, output_queue, proc_id, batch_size=32, subset='trainval', feat_type='res5c', version_suffix='', delta=0.2): super(AttentionDataPrefetcher, self).__init__() self._batch_size = batch_size self._proc_id = proc_id self._num_top_ans = 2000 self._queue = output_queue self._version_suffix = version_suffix self._subset = subset self._delta = delta self._images = None self._quest_len = None self._quest = None self._answer = None self.quest_index2counter_index = None self._num = None self._valid_ids = None self._load_data() self._feat_type = feat_type.lower() self._FEAT_ROOT = get_feature_root(self._subset, self._feat_type) self._transpose_feat = 'res5c' in self._feat_type
def __init__(self, subset='kprestval'): self._subset = subset self._FEAT_ROOT = get_feature_root(self._subset, 'res5c') self._load_data()