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)