原因分析
- 使用以下命令设置日志级别并设置环境变量将ATC日志打印到屏幕。
--log=error export ASCEND_SLOG_PRINT_TO_STDOUT=1 export ASCEND_GLOBAL_LOG_LEVEL=0
- 重新运行ATC,日志打印信息如下图所示。
根据日志打印信息,发现Resize_1算子报错。
- 使用Netron软件打开Resize_1算子,如下图所示。
由于Resize算子输入为空,导致无法转OM模型。
解决措施
补全Resize算子的input信息。
ATC转OM模型时Resize算子的input信息不能为空,补全Resize算子的inputs信息,单击链接下载 auto_optimizer。
from auto_optimizer import OnnxGraph
import numpy as np
model = OnnxGraph.parse('RetinaFace.onnx')
model.add_initializer('empty', np.array([]))
for node in model.get_nodes('Resize'):
print(node)
node.inputs[1] = 'empty'
node.inputs[2] = 'empty'
print(node)
model.save('RetinaFace_add_inputs.onnx')




