コード例 #1
0
 def test_dynamic_init(self):
     sampler = WeightedSampler(reader=get_dynamic_window_reader(),
                               data_param=DYNAMIC_MOD_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)
         with self.assertRaisesRegexp(tf.errors.OutOfRangeError, ""):
             out = sess.run(sampler.pop_batch_op())
コード例 #2
0
ファイル: sampler_weighted_test.py プロジェクト: amh28/NIF
 def test_dynamic_init(self):
     sampler = WeightedSampler(reader=get_dynamic_window_reader(),
                               data_param=DYNAMIC_MOD_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)
         with self.assertRaisesRegexp(tf.errors.OutOfRangeError, ""):
             out = sess.run(sampler.pop_batch_op())
コード例 #3
0
 def test_dynamic_init(self):
     sampler = WeightedSampler(reader=get_dynamic_window_reader(),
                               data_param=DYNAMIC_MOD_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[1:], (8, 2, 256, 2))
コード例 #4
0
 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()
コード例 #5
0
ファイル: sampler_weighted_test.py プロジェクト: amh28/NIF
 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()