1. 18 Jul, 2017 3 commits
    • Derek Chow's avatar
      Enable inference with dynamic batch size in SSD. · 4f14cb62
      Derek Chow authored
      * Creates a new batch_decode method in SSD Meta architecture that can handle
        dynamic batch size.
      * use combined_shapes in _get_feature_maps_spatial_dims method to handle
        dynamic batch image_size.
      * Add dynamic batch size tests to check preprocess, predict and postprocess
        methods in SSD Meta architecture.
      4f14cb62
    • 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
    • Derek Chow's avatar
  2. 15 Jun, 2017 1 commit