Update Exporter for `tf_keras` install (#13231)

6 months ago · 6b28bb6e48
parent 1a40d30dbf
commit 6b28bb6e48
1 changed files with 2 additions and 1 deletions
--- a/ultralytics/engine/exporter.py
+++ b/ultralytics/engine/exporter.py
@ -819,7 +819,8 @@ class Exporter:
            check_requirements("cmake")  # 'cmake' is needed to build onnxsim on aarch64
        check_requirements(
            (
-                "keras",
+                "keras",  # required by onnx2tf package
+                "tf_keras",  # required by onnx2tf package
                "onnx>=1.12.0",
                "onnx2tf>1.17.5,<=1.22.3",
                "sng4onnx>=1.0.1",