ONNX模型转换RKNN模型失败

[yanguiren122333]

报错信息如下： W:tensorflow:From E:\tools\anaconda\Anaconda_envs\envs\py36\lib\site-packages\rknn\api\rknn.py:150: The name tf.enable_eager_execution is deprecated. Please use tf.compat.v1.enable_eager_execution instead.

2024-07-17 11:05:14.298500: I tensorflow/core/platform/cpu_feature_guard.cc:142] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 E Calc node NonMaxSuppression : NonMaxSuppression_405 output shape fail E Catch exception when loading onnx model: mmdeploy_models/mmdet/yolov3\end2end.onnx! E Traceback (most recent call last): E File "rknn\base\RKNNlib\onnx_ir\onnx_numpy_backend\shape_inference.py", line 65, in rknn.base.RKNNlib.onnx_ir.onnx_numpy_backend.shape_inference.infer_shape E File "rknn\base\RKNNlib\onnx_ir\onnx_numpy_backend\ops\nonmaxsuppression.py", line 18, in rknn.base.RKNNlib.onnx_ir.onnx_numpy_backend.ops.nonmaxsuppression.NonMaxSuppression E File "E:\tools\anaconda\Anaconda_envs\envs\py36\lib\site-packages\tensorflow\python\ops\image_ops_impl.py", line 2587, in non_max_suppression E iou_threshold, score_threshold) E File "E:\tools\anaconda\Anaconda_envs\envs\py36\lib\site-packages\tensorflow\python\ops\gen_image_ops.py", line 2759, in non_max_suppression_v3 E _six.raise_from(_core._status_to_exception(e.code, message), None) E File "", line 3, in raise_from E tensorflow.python.framework.errors_impl.InvalidArgumentError: scores must be 1-D[80,5000] [Op:NonMaxSuppressionV3] E During handling of the above exception, another exception occurred: E Traceback (most recent call last): E File "rknn\base\RKNNlib\app\importer\import_onnx.py", line 118, in rknn.base.RKNNlib.app.importer.import_onnx.Importonnx.run E File "rknn\base\RKNNlib\converter\convert_onnx.py", line 112, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.init E File "rknn\base\RKNNlib\converter\convert_onnx.py", line 1062, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx._shape_inference E File "rknn\base\RKNNlib\onnx_ir\onnx_numpy_backend\shape_inference.py", line 70, in rknn.base.RKNNlib.onnx_ir.onnx_numpy_backend.shape_inference.infer_shape E File "rknn\api\rknn_log.py", line 323, in rknn.api.rknn_log.RKNNLog.e E ValueError: Calc node NonMaxSuppression : NonMaxSuppression_405 output shape fail E Please feedback the detailed log file <log_feedback_to_the_rknn_toolkit_dev_team.log> to the RKNN Toolkit development team. E You can also check github issues: https://github.com/rockchip-linux/rknn-toolkit/issues 07/17 11:05:14 - mmengine - ERROR - E:\tools\anaconda\Anaconda_envs\envs\py36\lib\site-packages\mmdeploy\backend\rknn\onnx2rknn.py - onnx2rknn - 86 - Load model failed!