1. 18 Jul, 2017 7 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
      Changes to Batch Non-Max Suppression to enable batch inference. · 4d641f7f
      Derek Chow authored
      A few change to prepare for batch inference:
      
      * Modify the return type of batch non max suppression to be tuple of tensors
        so it can be reused for both stages of faster rcnn without any confusion
        in the semantics implied the the keys used to represent the tensors.
      * Allow dynamic number of anchors (boxes) in addition to dynamic batch size.
      * Remove a redundant dynamic batch size test.
      4d641f7f
    • Derek Chow's avatar
      Support dynamic batch size in batch_multiclass_non_max_suppression. · 13c46302
      Derek Chow authored
      This change is required to enable object detection export and inference
      with dynamic batch size.
      13c46302
    • Derek Chow's avatar
    • Derek Chow's avatar
      Rename Saver used in init_fn · ec91f6b8
      Derek Chow authored
      Bug:
      We use two different Saver objects during training, one to initialize from a
      checkpoint and another to save and restore from checkpoints during training.
      Both these Savers were bound to same variable name causing init_fn function
      to use the latter instead of the former when called.
      ec91f6b8
    • Derek Chow's avatar
  2. 17 Jul, 2017 4 commits
  3. 15 Jul, 2017 1 commit
  4. 14 Jul, 2017 3 commits
  5. 13 Jul, 2017 12 commits
  6. 12 Jul, 2017 4 commits
  7. 11 Jul, 2017 6 commits
  8. 10 Jul, 2017 3 commits