本文整理匯總了Python中object_detection.utils.ops.retain_groundtruth方法的典型用法代碼示例。如果您正苦於以下問題：Python ops.retain_groundtruth方法的具體用法？Python ops.retain_groundtruth怎麽用？Python ops.retain_groundtruth使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類object_detection.utils.ops的用法示例。
在下文中一共展示了ops.retain_groundtruth方法的2個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: test_filter_with_missing_fields
# 需要導入模塊: from object_detection.utils import ops [as 別名]
# 或者: from object_detection.utils.ops import retain_groundtruth [as 別名]
def test_filter_with_missing_fields(self):
input_boxes = tf.placeholder(tf.float32, shape=(None, 4))
input_classes = tf.placeholder(tf.int32, shape=(None,))
input_tensors = {
fields.InputDataFields.groundtruth_boxes: input_boxes,
fields.InputDataFields.groundtruth_classes: input_classes
}
valid_indices = tf.placeholder(tf.int32, shape=(None,))
feed_dict = {
input_boxes:
np.array([[0.2, 0.4, 0.1, 0.8], [0.2, 0.4, 1.0, 0.8]], dtype=np.float),
input_classes:
np.array([1, 2], dtype=np.int32),
valid_indices:
np.array([0], dtype=np.int32)
}
expected_tensors = {
fields.InputDataFields.groundtruth_boxes:
[[0.2, 0.4, 0.1, 0.8]],
fields.InputDataFields.groundtruth_classes:
[1]
}
output_tensors = ops.retain_groundtruth(input_tensors, valid_indices)
with self.test_session() as sess:
output_tensors = sess.run(output_tensors, feed_dict=feed_dict)
for key in [fields.InputDataFields.groundtruth_boxes]:
self.assertAllClose(expected_tensors[key], output_tensors[key])
for key in [fields.InputDataFields.groundtruth_classes]:
self.assertAllEqual(expected_tensors[key], output_tensors[key]) 示例2: test_filter_with_empty_groundtruth_boxes
# 需要導入模塊: from object_detection.utils import ops [as 別名]
# 或者: from object_detection.utils.ops import retain_groundtruth [as 別名]
def test_filter_with_empty_groundtruth_boxes(self):
input_boxes = tf.placeholder(tf.float32, shape=(None, 4))
input_classes = tf.placeholder(tf.int32, shape=(None,))
input_is_crowd = tf.placeholder(tf.bool, shape=(None,))
input_area = tf.placeholder(tf.float32, shape=(None,))
input_difficult = tf.placeholder(tf.float32, shape=(None,))
valid_indices = tf.placeholder(tf.int32, shape=(None,))
input_tensors = {
fields.InputDataFields.groundtruth_boxes: input_boxes,
fields.InputDataFields.groundtruth_classes: input_classes,
fields.InputDataFields.groundtruth_is_crowd: input_is_crowd,
fields.InputDataFields.groundtruth_area: input_area,
fields.InputDataFields.groundtruth_difficult: input_difficult
}
output_tensors = ops.retain_groundtruth(input_tensors, valid_indices)
feed_dict = {
input_boxes:
np.array([], dtype=np.float).reshape(0, 4),
input_classes:
np.array([], dtype=np.int32),
input_is_crowd:
np.array([], dtype=np.bool),
input_area:
np.array([], dtype=np.float32),
input_difficult:
np.array([], dtype=np.float32),
valid_indices:
np.array([], dtype=np.int32)
}
with self.test_session() as sess:
output_tensors = sess.run(output_tensors, feed_dict=feed_dict)
for key in input_tensors:
if key == fields.InputDataFields.groundtruth_boxes:
self.assertAllEqual([0, 4], output_tensors[key].shape)
else:
self.assertAllEqual([0], output_tensors[key].shape)