• Pooya Davoodi's avatar
    Add Combined NMS (#6138) · 3980d2a1
    Pooya Davoodi authored
    * Updating python API to use CombinedNonMaxSuppresion TF operator
    
    1. Adds a unit test to test post_processing python API
    2. Currently sets clip_window to None as the kernel uses the default
       clip_window of [0,0,1,1]
    3. Added use_static_shapes to the API. In old API if
       use_static_shapes is true, then it pads/clips outputs to max_total_size, if
    specified. If not specified, it pads to num_classes*max_size_per_class.
     If use_static_shapes is false, it always pads/clips to max_total_size.
    
    Update unit test to account for clipped bouding boxes
    
    Changed the name to CombinedNonMaxSuppression based on feedback from Google
    
    Added additional parameters to combinedNMS python function. They are currently
    unused and required for networks like FasterRCNN and MaskRCNN
    
    * Delete selected_indices from API
    
    Because it was removed from CombinedNMS recently in the PR.
    
    * Improve doc of function combined_non_max_suppression
    
    * Enable CombinedNonMaxSuppression for first_stage_nms
    
    * fix bug
    
    * Ensure agnostic_nms is not used with combined_nms
    
    Remove redundant arguments from combined_nms
    
    * Fix pylint
    
    * Add checks for unsupported args
    
    * Fix pylint
    
    * Move combined_non_max_suppression to batch_multiclass_non_max_suppression
    
    Also rename combined_nms to use_combined_nms
    
    * Delete combined_nms for first_stage_nms because it does not work
    
    * Revert "Delete combined_nms for first_stage_nms because it does not work"
    
    This reverts commit 2a3cc5145f17cee630a67ddedd20e90c2920fa9f.
    
    * Use nmsed_additional_fields.get to avoid error
    
    * Merge combined_non_max_suppression with main nms function
    
    * Rename combined_nms for first stage nms
    
    * Improve  docs
    
    * Use assertListEqual for numpy arrays
    
    * Fix pylint errors
    
    * End comments with period
    3980d2a1
post_processing.proto 2.71 KB