def __init__(self,
queue,
sketch_dir,
image_dir,
triplet_path,
mean,
hard_ratio,
batch_size,
phase,
proc_id,
mode,
att_flag=False):
"""Setup the TripletSamplingDataLayer."""
super(TripletSamplingDataFetcher, self).__init__()
# mean = mean
self._queue = queue
self._phase = phase
self.proc_id = proc_id
self.sketch_transformer = Transformer(225, 1, mean,
self._phase == "TRAIN")
self.anc_bm = MemoryBlockManager(sketch_dir)
self.disable_anc_id = 1484
self.valid_anc_inds = np.setdiff1d(np.arange(self.anc_bm.num_samples),
np.array(self.disable_anc_id))
self.pos_neg_bm = MemoryBlockManager(image_dir)
self.hard_ratio = hard_ratio
self.mode = mode
self.mini_batchsize = batch_size
self.att_flag = att_flag
self.load_triplets(triplet_path)
# self.tmp_flag = True
# self.idx = 0
print 'Done'
def __init__(self, queue, sketch_data, image_data, sketch_label,
image_label, triplets, queue_paras, phase):
"""Setup the TripletSamplingDataLayer."""
super(TripletSamplingDataFetcher, self).__init__()
# mean = mean
self.im_size = queue_paras['im_size']
self.cp_size = queue_paras['cp_size']
self.chns = queue_paras['chns']
self.mean = queue_paras['mean']
self.batchsize = queue_paras['batch_size']
self.num_epoch = queue_paras['num_epochs']
self._queue = queue
self._phase = phase
self.sketch_transformer = Transformer(self.cp_size, self.chns,
self.mean,
self._phase == "TRAIN")
if sketch_label is None or image_label is None:
self.has_label = False
else:
self.has_label = True
self.anc_bm = MemoryBlockManager(sketch_data, sketch_label,
self.has_label)
self.pos_neg_bm = MemoryBlockManager(image_data, image_label,
self.has_label)
self.triplets = triplets
def __init__(self, queue, sketch_dir, mean, hard_ratio, batch_size):
"""Setup the StrokeSamplingDataLayer."""
super(SamplingDataFetcher, self).__init__()
# mean = mean
self._queue = queue
self.sketch_transformer = Transformer(225, 1, mean)
self.sketch_dir = sketch_dir
self.sketch_bm = MemoryBlockManager(sketch_dir)
self.hard_ratio = hard_ratio
self.mini_batchsize = batch_size
def __init__(self, queue, sketch_dir, image_dir, triplet_path, mean, hard_ratio, batch_size, phase):
"""Setup the TripletSamplingDataLayer."""
super(TripletSamplingDataFetcher, self).__init__()
# mean = mean
self._queue = queue
self._phase = phase
self.sketch_transformer = Transformer(225, 1, mean, self._phase == "TRAIN")
self.sketch_dir = sketch_dir
self.anc_bm = MemoryBlockManager(sketch_dir)
self.pos_neg_bm = MemoryBlockManager(image_dir)
self.hard_ratio = hard_ratio
self.mini_batchsize = batch_size
self.load_triplets(triplet_path)
def __init__(self, queue, layer_params):
"""Setup the TripletSamplingDataLayer."""
super(TripletSamplingDataFetcher, self).__init__()
self._queue = queue
mean = layer_params['mean']
self._phase = layer_params['phase']
self.sketch_transformer = Transformer(225, 1, mean,
self._phase == "TRAIN")
self.anc_bm = MemoryBlockManager(layer_params['sketch_dir'])
self.pos_neg_bm = MemoryBlockManager(layer_params['image_dir'])
self.hard_ratio = layer_params['hard_ratio']
self.mini_batchsize = layer_params['batch_size']
self.load_triplets(layer_params['triplet_path'])
