• 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
test_utils.py 4.47 KB