• pkulzc's avatar
    Minor fixes for object detection. · 1f484095
    pkulzc authored
    214018767  by Zhichao Lu:
    
        Add original_image_spatial_shape tensor in input dictionary to store shape of the original input image
    
    --
    213914693  by lzc:
    
        Internal change.
    
    --
    213872175  by Zhichao Lu:
    
        This CL adds a Keras-based mobilenet_v2 feature extractor for object detection models.
    
        As part of this CL, we use the Keras mobilenet_v2 application's keyword argument layer injection API to allow the generated network to support the object detection hyperparameters.
    
    --
    213848499  by Zhichao Lu:
    
        Replace tf.image.resize_nearest_neighbor with tf.image.resize_images. tf.image.resize_nearest_neighbor only supports 4-D tensors but masks is a 3-D tensor.
    
    --
    213758622  by lzc:
    
        Internal change.
    
    --
    
    PiperOrigin-RevId: 214018767
    1f484095
test_case.py 4.14 KB