onnx-tf
-
How to convert pt to tflite (yolov5 to tensorflow lite) tflite로 변환AI 2021. 1. 19. 16:18
업데이트: 이걸로 해결 github.com/zldrobit/yolov5/blob/tf-android/models/tf.py $ python models/tf.py --weights weights/best.pt --cfg models/yolov5s.yaml --img-size 320 yolov5로 학습된 모델을 1차적으로 Android에서 사용하고자 한다. 그럴러면 최종적으로 .pt 파일을 .tflite로 변환해야한다. 그게 한번에 되냐? 아쉽게도 pt -> onnx -> pb -> tflite 순으로 변환해야한다. 명령어 하나로 되었다면 얼마나 좋았을까.. 그래서 그 과정에서 꽤 많은 삽질을 하게 되었다. 잠깐 언급하자면 tensorflow 버전과 onnx 버전의 호환성.. 버전업되면서 바뀐것도 많고 알..