Python visualize_boxes_in_image Examples

Example #1

 def graph_fn():
   image = tf.zeros((6, 4, 3))
   corners = tf.constant([[0, 0, 5, 3],
                          [0, 0, 3, 2]], tf.float32)
   boxes = box_list.BoxList(corners)
   image_and_boxes = box_list_ops.visualize_boxes_in_image(image, boxes)
   image_and_boxes_bw = tf.cast(
       tf.greater(tf.reduce_sum(image_and_boxes, 2), 0.0), dtype=tf.float32)
   return image_and_boxes_bw

Example #2

File: box_list_ops_test.py Project: jonathangenty0801/Object-detection

 def test_visualize_boxes_in_image(self):
     image = tf.zeros((6, 4, 3))
     corners = tf.constant([[0, 0, 5, 3], [0, 0, 3, 2]], tf.float32)
     boxes = box_list.BoxList(corners)
     image_and_boxes = box_list_ops.visualize_boxes_in_image(image, boxes)
     image_and_boxes_bw = tf.to_float(
         tf.greater(tf.reduce_sum(image_and_boxes, 2), 0.0))
     exp_result = [[1, 1, 1, 0], [1, 1, 1, 0], [1, 1, 1, 0], [1, 0, 1, 0],
                   [1, 1, 1, 0], [0, 0, 0, 0]]
     with self.test_session() as sess:
         output = sess.run(image_and_boxes_bw)
         self.assertAllEqual(output.astype(int), exp_result)

Example #3

File: box_list_ops_test.py Project: Exscotticus/models

 def test_visualize_boxes_in_image(self):
   image = tf.zeros((6, 4, 3))
   corners = tf.constant([[0, 0, 5, 3],
                          [0, 0, 3, 2]], tf.float32)
   boxes = box_list.BoxList(corners)
   image_and_boxes = box_list_ops.visualize_boxes_in_image(image, boxes)
   image_and_boxes_bw = tf.to_float(
       tf.greater(tf.reduce_sum(image_and_boxes, 2), 0.0))
   exp_result = [[1, 1, 1, 0],
                 [1, 1, 1, 0],
                 [1, 1, 1, 0],
                 [1, 0, 1, 0],
                 [1, 1, 1, 0],
                 [0, 0, 0, 0]]
   with self.test_session() as sess:
     output = sess.run(image_and_boxes_bw)
     self.assertAllEqual(output.astype(int), exp_result)