def test_dynamic_window_enqueue(self): enqueue_size = 3 dequeue_size = 2 test_buffer = InputBatchQueueRunner(capacity=10, shuffle=False) window = get_dynamic_image_window() test_buffer._create_queue_and_ops(window, enqueue_size, dequeue_size) dynamic_image_size = { 'image': (5, 5, 5, 1, 2), 'label': (5, 5, 3, 1, 2) } window_shape = window.match_image_shapes(dynamic_image_size) enqueue_dict = {} placeholder = window.image_data_placeholder('image') enqueue_dict[placeholder] = np.zeros((1, ) + window_shape['image']) placeholder = window.coordinates_placeholder('image') enqueue_dict[placeholder] = np.zeros(placeholder.shape.as_list()) placeholder = window.image_data_placeholder('label') enqueue_dict[placeholder] = np.zeros((1, ) + window_shape['label']) placeholder = window.coordinates_placeholder('label') enqueue_dict[placeholder] = np.zeros(placeholder.shape.as_list()) with self.test_session() as sess: # queue size before enqueue expected_queue_size = 0 queue_size = sess.run(test_buffer._query_queue_size_op) self.assertAllEqual(expected_queue_size, queue_size) # do enqueue sess.run(test_buffer._enqueue_op, feed_dict=enqueue_dict) # queue size after enqueue expected_queue_size = 1 queue_size = sess.run(test_buffer._query_queue_size_op) self.assertAllEqual(expected_queue_size, queue_size) # do dequeue sess.run(test_buffer.pop_batch_op()) # queue size after enqueue queue_size = sess.run(test_buffer._query_queue_size_op) self.assertAllEqual(0, queue_size) sess.run(test_buffer._close_queue_op)
def test_dynamic_window_enqueue(self): enqueue_size = 3 dequeue_size = 2 test_buffer = InputBatchQueueRunner(capacity=10, shuffle=False) window = get_dynamic_image_window() test_buffer._create_queue_and_ops(window, enqueue_size, dequeue_size) dynamic_image_size = {'image': (5, 5, 5, 1, 2), 'label': (5, 5, 3, 1, 2)} window_shape = window.match_image_shapes(dynamic_image_size) enqueue_dict = {} placeholder = window.image_data_placeholder('image') enqueue_dict[placeholder] = np.zeros((1,) + window_shape['image']) placeholder = window.coordinates_placeholder('image') enqueue_dict[placeholder] = np.zeros(placeholder.shape.as_list()) placeholder = window.image_data_placeholder('label') enqueue_dict[placeholder] = np.zeros((1,) + window_shape['label']) placeholder = window.coordinates_placeholder('label') enqueue_dict[placeholder] = np.zeros(placeholder.shape.as_list()) with self.test_session() as sess: # queue size before enqueue expected_queue_size = 0 queue_size = sess.run(test_buffer._query_queue_size_op) self.assertAllEqual(expected_queue_size, queue_size) # do enqueue sess.run(test_buffer._enqueue_op, feed_dict=enqueue_dict) # queue size after enqueue expected_queue_size = 1 queue_size = sess.run(test_buffer._query_queue_size_op) self.assertAllEqual(expected_queue_size, queue_size) # do dequeue sess.run(test_buffer.pop_batch_op()) # queue size after enqueue queue_size = sess.run(test_buffer._query_queue_size_op) self.assertAllEqual(0, queue_size) sess.run(test_buffer._close_queue_op)
def test_static_window_enqueue(self): enqueue_size = 3 dequeue_size = 2 test_buffer = InputBatchQueueRunner(capacity=10, shuffle=True) window = get_static_image_window() test_buffer._create_queue_and_ops(window, enqueue_size=enqueue_size, dequeue_size=dequeue_size) enqueue_dict = {} placeholder = window.image_data_placeholder('image') enqueue_dict[placeholder] = np.zeros(placeholder.shape.as_list()) placeholder = window.coordinates_placeholder('image') enqueue_dict[placeholder] = np.zeros(placeholder.shape.as_list()) placeholder = window.image_data_placeholder('label') enqueue_dict[placeholder] = np.zeros(placeholder.shape.as_list()) placeholder = window.coordinates_placeholder('label') enqueue_dict[placeholder] = np.zeros(placeholder.shape.as_list()) with self.test_session() as sess: # queue size before enqueue queue_size = sess.run(test_buffer._query_queue_size_op) self.assertAllEqual(0, queue_size) # do enqueue sess.run(test_buffer._enqueue_op, feed_dict=enqueue_dict) # queue size after enqueue queue_size = sess.run(test_buffer._query_queue_size_op) self.assertAllEqual(enqueue_size, queue_size) for _ in range(2): sess.run(test_buffer._enqueue_op, feed_dict=enqueue_dict) queue_size = sess.run(test_buffer._query_queue_size_op) self.assertAllEqual(9, queue_size) sess.run(test_buffer.pop_batch_op()) queue_size = sess.run(test_buffer._query_queue_size_op) self.assertAllEqual(7, queue_size) sess.run(test_buffer._close_queue_op)
def test_static_window_enqueue(self): enqueue_size = 3 dequeue_size = 2 test_buffer = InputBatchQueueRunner(capacity=10, shuffle=True) window = get_static_image_window() test_buffer._create_queue_and_ops(window, enqueue_size=enqueue_size, dequeue_size=dequeue_size) enqueue_dict = {} placeholder = window.image_data_placeholder('image') enqueue_dict[placeholder] = np.zeros(placeholder.shape.as_list()) placeholder = window.coordinates_placeholder('image') enqueue_dict[placeholder] = np.zeros(placeholder.shape.as_list()) placeholder = window.image_data_placeholder('label') enqueue_dict[placeholder] = np.zeros(placeholder.shape.as_list()) placeholder = window.coordinates_placeholder('label') enqueue_dict[placeholder] = np.zeros(placeholder.shape.as_list()) with self.test_session() as sess: # queue size before enqueue queue_size = sess.run(test_buffer._query_queue_size_op) self.assertAllEqual(0, queue_size) # do enqueue sess.run(test_buffer._enqueue_op, feed_dict=enqueue_dict) # queue size after enqueue queue_size = sess.run(test_buffer._query_queue_size_op) self.assertAllEqual(enqueue_size, queue_size) for _ in range(2): sess.run(test_buffer._enqueue_op, feed_dict=enqueue_dict) queue_size = sess.run(test_buffer._query_queue_size_op) self.assertAllEqual(9, queue_size) sess.run(test_buffer.pop_batch_op()) queue_size = sess.run(test_buffer._query_queue_size_op) self.assertAllEqual(7, queue_size) sess.run(test_buffer._close_queue_op)