handle trt8 resize align corners api
@@ -86,6 +86,10 @@ def convert_interpolate_trt7(ctx):
layer.resize_mode=trt.ResizeMode.NEAREST
if align_corners != None:
+ if trt_version() > '8.0':
+ if align_corners:
+ layer.coordinate_transformation = trt.ResizeCoordinateTransformation.ALIGN_CORNERS
+ else:
layer.align_corners = align_corners
output._trt = layer.get_output(0)
