def test_close_early(self):
sampler = WeightedSampler(reader=get_2d_reader(),
data_param=MOD_2D_DATA,
batch_size=2,
windows_per_image=10,
queue_length=10)
sampler.close_all()
def test_close_early(self):
sampler = WeightedSampler(reader=get_2d_reader(),
data_param=MOD_2D_DATA,
batch_size=2,
windows_per_image=10,
queue_length=10)
sampler.close_all()
def test_2d_init(self):
sampler = WeightedSampler(reader=get_2d_reader(),
data_param=MOD_2D_DATA,
batch_size=2,
windows_per_image=10,
queue_length=10)
with self.test_session() as sess:
coordinator = tf.train.Coordinator()
sampler.run_threads(sess, coordinator, num_threads=2)
out = sess.run(sampler.pop_batch_op())
self.assertAllClose(out['image'].shape, (2, 10, 9, 1))
sampler.close_all()
def test_2d_init(self):
sampler = WeightedSampler(reader=get_2d_reader(),
data_param=MOD_2D_DATA,
batch_size=2,
windows_per_image=10,
queue_length=10)
with self.test_session() as sess:
coordinator = tf.train.Coordinator()
sampler.run_threads(sess, coordinator, num_threads=2)
out = sess.run(sampler.pop_batch_op())
self.assertAllClose(out['image'].shape, (2, 10, 9, 1))
sampler.close_all()