-
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