Documento de problemas al convertir el modelo resnet50 basado en mxnet a la implementación de ONNX

Este artículo registra dos errores y soluciones al convertir el modelo resnet50 entrenado en mxnet al formato onnx para su implementación

Sistema operativo: ubuntu 16.04

Mxnet: 1.6.0

onnx: 1.6.0

cuda: 10.2

La capa de especificación de lotes utilizada en resnet50 se convirtió a ONNX. 0

La capa de especificación por lotes utilizada en resnet50 informó un error al convertir a ONNX que indica que los atributos espaciales no son compatibles.

mxnet source mxnet/contrib/onnx/mx2onnx/_op_translation py line 359. cancela el registro de los parámetros espaciales

Verifique la estructura de la red a través de netron.app y descubra que el parámetro bn_data_gamma del primer lote de capas estándar (bn_data) es anormal:

Cambie manualmente la capa bn_data en el modelo onnx, el parámetro bn_data_gamma se cambia a 1.0

.