yolo export - TypeError: You are passing KerasTensor 오류 해결

해당 오류는 yolo export 시 Ubuntu 18 환경에서 발생한 오류입니다.

저의 경우는 yolo pose 모델의 keypoint 수를 17개에서 10개로 더 줄여서 custom model로 학습했는데 그것과는 상관없이 환경 차이로 발생한 걸로 보입니다.

 

오류 전체 구문은 아래와 같습니다.

TypeError: You are passing KerasTensor(type_spec=TensorSpec(shape=(1, 20, 20, 576), dtype=tf.float32, name=None), name='tf.math.multiply_179/Mul:0', description="created by layer 'tf.math.multiply_179'"), an intermediate Keras symbolic input/output, to a TF API that does not allow registering custom dispatchers, such as `tf.cond`, `tf.function`, gradient tapes, or `tf.map_fn`. Keras Functional model construction only supports TF API calls that *do* support dispatching, such as `tf.math.add` or `tf.reshape`. Other APIs cannot be called directly on symbolic Kerasinputs/outputs. You can work around this limitation by putting the operation in a custom Keras layer `call` and calling that layer on this symbolic input/output.

 

stackoverflow 등을 보면 python 3.9로 업데이트해라, 라이브러리 버전 맞춰라 등 많은데 결국 안됐고, colab 환경에서 성공했습니다.

아래 url의 colab 복사해서 export 부분 실행하시면 정상적으로 변환됩니다.

https://colab.research.google.com/github/ultralytics/ultralytics/blob/main/examples/tutorial.ipynb

Google Colab Notebook