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
Exemplo n.º 3
0
 def __init__(self, subset='kprestval'):
     self._subset = subset
     self._FEAT_ROOT = get_feature_root(self._subset, 'res5c')
     self._load_data()