experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Shift forward-compat window for NMS padded to fix TFLite conversion in 2.3 

PiperOrigin-RevId: 319856402
Change-Id: I60eeb4e411609511046cc4676c2f3ad630b3769d
This commit is contained in:
Sachin Joglekar 2020-07-06 14:30:18 -07:00
parent fc9d68a5e8
commit 1725ab6962
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/ops/image_ops_impl.py
View File

@ -4579,11 +4579,11 @@ def non_max_suppression_padded(boxes,
Raises: Raises:
ValueError: When set pad_to_max_output_size to False for batched input. ValueError: When set pad_to_max_output_size to False for batched input.
""" """
# if no new arguments are used and no later than 2020/4/20, use the old # if no new arguments are used and no later than 2020/6/23, use the old
# version to give us time to fix TFLite conversion # version to give us time to fix TFLite conversion after the TF 2.3 release.
if (not sorted_input) and \ if (not sorted_input) and \
(not canonicalized_coordinates) and \ (not canonicalized_coordinates) and \
tile_size == 512 and not compat.forward_compatible(2020, 4, 20): tile_size == 512 and not compat.forward_compatible(2020, 6, 23):
return non_max_suppression_padded_v1( return non_max_suppression_padded_v1(
boxes, scores, max_output_size, iou_threshold, score_threshold, boxes, scores, max_output_size, iou_threshold, score_threshold,
pad_to_max_output_size, name) pad_to_max_output_size, name)