• Derek Chow's avatar
    Enable inference with dynamic batch size in Faster RCNN. · 5d5fb7cc
    Derek Chow authored
    * Adds a util function to compute a mix of dynamic and static shapes
      preferring static when available.
    * Uses batch_multiclass_non_max_suppression function in postprocess_rpn
      instead of looping over static batch shape and performing
      multiclass_non_max_suppression.
    * Adds a new helper function _unpad_proposals_and_sample_boxclassifier_batch
      to sample from a batch of tensors possibly containing paddings.
    * Tests batch inference with various configurations of static shape via
      unittests.
    5d5fb7cc
shape_utils_test.py 5.05 KB