class SamplingDataFetcher(Process):
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 get_next_batch(self):
sketch_batch = []
# sampling
sketch_inds = self.sketch_bm.pop_batch_inds_circular(self.mini_batchsize)
# fetch data
for (sketch_id) in zip(sketch_inds):
sketch_batch.append(self.sketch_bm.get_sample(sketch_id).reshape((256, 256, 1)))
# apply transform
sketch_batch = self.sketch_transformer.transform_all(sketch_batch).astype(np.uint8)
self._queue.put(sketch_batch)
def run(self):
print ('SamplingDataFetcher started')
while True:
self.get_next_batch()
class TripletSamplingDataFetcher_lot(Process):
def __init__(self, queue, sketch_dir, dir1, dir2, triplet_path, mean, hard_ratio, batch_size, phase):
"""Setup the TripletSamplingDataLayer."""
super(TripletSamplingDataFetcher_lot, 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_bm1 = MemoryBlockManager(dir1)
self.pos_neg_bm2 = MemoryBlockManager(dir2)
self.hard_ratio = hard_ratio
self.mini_batchsize = batch_size
self.load_triplets(triplet_path)
def load_triplets(self, triplet_path):
self.triplets, self.neg_list = load_triplets(triplet_path, self._phase)
def get_next_batch(self):
anc_batch = []; pos_batch1 = []; neg_batch1 = []; pos_batch2 = []; neg_batch2 = [];
# sampling
anc_inds = self.anc_bm.pop_batch_inds_circular(self.mini_batchsize)
if 'handbags' in self.sketch_dir:
# positive are always true match
pos_inds, neg_inds = sample_triplets_trueMatch(anc_inds, self._phase)
else:
pos_inds, neg_inds = sample_triplets_pos_neg(anc_inds, self.triplets, self.neg_list, self.hard_ratio)
# fetch data
for (anc_id, pos_id, neg_id) in zip(anc_inds, pos_inds, neg_inds):
anc_batch.append(self.anc_bm.get_sample(anc_id).reshape((256, 256, 1)))
pos_batch1.append(self.pos_neg_bm1.get_sample(pos_id).reshape((256, 256, 1)))
neg_batch1.append(self.pos_neg_bm1.get_sample(neg_id).reshape((256, 256, 1)))
pos_batch2.append(self.pos_neg_bm2.get_sample(pos_id).reshape((256, 256, 1)))
neg_batch2.append(self.pos_neg_bm2.get_sample(neg_id).reshape((256, 256, 1)))
# apply transform
anc_batch = self.sketch_transformer.transform_all(anc_batch).astype(np.uint8)
pos_batch1 = self.sketch_transformer.transform_all(pos_batch1).astype(np.uint8)
neg_batch1 = self.sketch_transformer.transform_all(neg_batch1).astype(np.uint8)
pos_batch2 = self.sketch_transformer.transform_all(pos_batch2).astype(np.uint8)
neg_batch2 = self.sketch_transformer.transform_all(neg_batch2).astype(np.uint8)
self._queue.put((anc_batch, pos_batch1, neg_batch1, pos_batch2, neg_batch2))
def run(self):
print ('TripletSamplingDataFetcher started')
while True:
self.get_next_batch()
class TripletSamplingDataFetcher(Process):
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'])
def load_triplets(self, triplet_path):
self.triplets, self.neg_list = load_triplets(triplet_path, self._phase)
def get_next_batch(self):
anc_batch = []
pos_batch = []
neg_batch = []
# sampling
anc_inds = self.anc_bm.pop_batch_inds_circular(self.mini_batchsize)
pos_inds, neg_inds = sample_triplets(anc_inds, self.triplets,
self.neg_list, self.hard_ratio)
# fetch data
for (anc_id, pos_id, neg_id) in zip(anc_inds, pos_inds, neg_inds):
anc_batch.append(
self.anc_bm.get_sample(anc_id).reshape((1, 256, 256)))
pos_batch.append(
self.pos_neg_bm.get_sample(pos_id).reshape((1, 256, 256)))
neg_batch.append(
self.pos_neg_bm.get_sample(neg_id).reshape((1, 256, 256)))
# apply transform
anc_batch = self.sketch_transformer.transform_all(anc_batch)
pos_batch = self.sketch_transformer.transform_all(pos_batch)
neg_batch = self.sketch_transformer.transform_all(neg_batch)
self._queue.put((anc_batch, pos_batch, neg_batch))
def run(self):
print 'TripletSamplingDataFetcher started'
while True:
self.get_next_batch()
class TripletSamplingLayer(object):
def __init__(self, sketch_dir, image_dir, triplet_path, mean, hard_ratio,
batch_size, phase):
"""Setup the TripletSamplingDataLayer."""
self._queue = Queue(10)
# mean = mean
self._phase = phase
self.sketch_transformer = Transformer(225, 1, mean,
self._phase == "TRAIN")
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 load_triplets(self, triplet_path):
self.triplets, self.neg_list = load_triplets(triplet_path, self._phase)
def get_next_batch(self):
anc_batch = []
pos_batch = []
neg_batch = []
# sampling
anc_inds = self.anc_bm.pop_batch_inds_circular(self.mini_batchsize)
pos_inds, neg_inds = sample_triplets(anc_inds, self.triplets,
self.neg_list, self.hard_ratio)
# fetch data
for (anc_id, pos_id, neg_id) in zip(anc_inds, pos_inds, neg_inds):
anc_batch.append(
self.anc_bm.get_sample(anc_id).reshape((256, 256, 1)))
pos_batch.append(
self.pos_neg_bm.get_sample(pos_id).reshape((256, 256, 1)))
neg_batch.append(
self.pos_neg_bm.get_sample(neg_id).reshape((256, 256, 1)))
# apply transform
anc_batch = self.sketch_transformer.transform_all(anc_batch)
pos_batch = self.sketch_transformer.transform_all(pos_batch)
neg_batch = self.sketch_transformer.transform_all(neg_batch)
# self._queue.put((anc_batch, pos_batch, neg_batch))
return anc_batch, pos_batch, neg_batch
class TripletSamplingDataFetcher(Process):
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'])
def load_triplets(self, triplet_path):
self.triplets, self.neg_list = load_triplets(triplet_path, self._phase)
def get_next_batch(self):
anc_batch = []; pos_batch = []; neg_batch = []
# sampling
anc_inds = self.anc_bm.pop_batch_inds_circular(self.mini_batchsize)
pos_inds, neg_inds = sample_triplets(anc_inds, self.triplets, self.neg_list, self.hard_ratio)
# fetch data
for (anc_id, pos_id, neg_id) in zip(anc_inds, pos_inds, neg_inds):
anc_batch.append(self.anc_bm.get_sample(anc_id).reshape((1, 256, 256)))
pos_batch.append(self.pos_neg_bm.get_sample(pos_id).reshape((1, 256, 256)))
neg_batch.append(self.pos_neg_bm.get_sample(neg_id).reshape((1, 256, 256)))
# apply transform
anc_batch = self.sketch_transformer.transform_all(anc_batch)
pos_batch = self.sketch_transformer.transform_all(pos_batch)
neg_batch = self.sketch_transformer.transform_all(neg_batch)
self._queue.put((anc_batch, pos_batch, neg_batch))
def run(self):
print 'TripletSamplingDataFetcher started'
while True:
self.get_next_batch()
