1. 模型輸入相關問題
圖像尺寸信息
- 模型輸入尺寸由原始模型決定,在轉換時固定
- 圖像尺寸信息是模型固有屬性,不是轉換時添加的
- 對于使用動態尺寸,可以在推理時自動根據當前的輸入尺寸推導輸出尺寸。
輸入格式(NCHW/NHWC) --input_format
- 不同框架默認格式不同:
- Caffe: 支持NCHW、ND(表示支持任意維度格式,N<=4)兩種格式,默認為NCHW。
- ONNX: 支持NCHW、NCDHW、ND(表示支持任意維度格式,N<=4)三種格式,默認為NCHW。
- TensorFlow: 支持NCHW、NHWC、ND、NCDHW、NDHWC五種輸入格式,默認為NHWC。
- MindSpore: 僅支持NCHW。
- 如果原始模型中含有帶data_format入參的算子,則該參數必填,推薦取值為ND,模型轉換過程中會根據data_format屬性的算子,推導出具體的format。若用戶無法確定輸入數據格式,則推薦指定為ND。
- 可通過
input_format
參數在轉換時指定
數據類型
- 默認輸入類型由原始模型決定
- 輸出默認為FP32,可通過
out_n