Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf.image.generate_bounding_box_proposals

View source on GitHub

Generate bounding box proposals from encoded bounding boxes.

tf.image.generate_bounding_box_proposals(
    scores, bbox_deltas, image_info, anchors, nms_threshold=0.7, pre_nms_topn=6000,
    min_size=16, post_nms_topn=300, name=None
)

Returns:

  • rois: Region of interest boxes sorted by their scores.
  • roi_probabilities: scores of the ROI boxes in the ROIs' tensor.