Resize

功能

调整输入tensor大小

边界

【输入】(ONNX版本为Opset v10)

2个输入

x:一个tensor,数据类型:unit8、int8、int16、int32、int64、float16、float、double

scales:与输入x的秩相等的数组

【输入】(ONNX版本为Opset v11/v12/v14/v15)

2个输入

x:一个tensor,数据类型:float16、float

sizes:输出tensor的size

【输出】

一个输出

y:缩放后的张量

【属性】(ONNX版本为Opset v10)

mode:string,插值算法,取值包括nearest、linear,默认值为nearest

【属性】(ONNX版本为Opset v11/v12/v14/v15)

coordinate_transformation_mode:string,定义缩放后图像与原图像的坐标转换,取值包括align_corners、asymmetric、tf_half_pixel_for_nn、tf_crop_and_resize、pytorch_half_pixel、half_pixel,默认值为half_pixel

cubic_coeff_a:三次插值系数,数据类型为float,默认值为-0.75

exclude_outside:超出tensor外的权重,数据类型为int,默认值为0

mode:string,插值算法,取值包括nearest、linear、cubic,默认值为nearest

nearest_mode:string,最近邻算子模式,取值包括round_prefer_floor、round_prefer_ceil、floor、ceil默认值为round_prefer_floor,本参数仅在mode为nearest才可用

【约束】

目前仅支持nearest和linear插值方式来处理图片,并且需要修改模型将输入scales或sizes由placeholder改为const类型,可以使用onnxsimplifier简化模型

支持的ONNX版本

Opset v10/v11/v12/v14/v15/v16/v17/v18