saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-27 20:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix symbolic_shape_infer for Resize with roi (#4426)

Browse source 
Should only apply roi when coordinate_transformation_mode == tf_crop_and_resize
This commit is contained in:
KeDengMS 2020-07-05 23:37:36 -07:00 committed by GitHub
parent 0d4a65eede
commit 77cf51b13c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
onnxruntime/core/providers/nuphar/scripts/symbolic_shape_infer.py
View file

@ -898,11 +898,15 @@ class SymbolicShapeInference:
if sizes is not None:
new_sympy_shape = [sympy.simplify(sympy.floor(s)) for s in sizes]
self._update_computed_dims(new_sympy_shape)
elif roi is not None and scales is not None:
elif scales is not None:
rank = len(scales)
assert len(roi) == 2*rank
roi_start = list(roi)[:rank]
roi_end = list(roi)[rank:]
if get_attribute(node, 'coordinate_transformation_mode') == 'tf_crop_and_resize':
assert len(roi) == 2*rank
roi_start = list(roi)[:rank]
roi_end = list(roi)[rank:]
else:
roi_start = [0]*rank
roi_end = [1]*rank
scales = list(scales)
new_sympy_shape = [sympy.simplify(sympy.floor(d * (end - start) * scale)) for d, start, end, scale in zip(input_sympy_shape, roi_start, roi_end, scales)]
self._update_computed_dims(new_sympy_shape)