본문 바로가기

끄적끄적

TFLite visualizing

오랜만의 포스팅이다.

TFLite를 사용해야 한다는 판단이 생겨, TFLite를 사용하고자 한다.

TFLite를 사용하더라도, TFLite로 재생성된 모델의 생김새, immediate output등을 알아야 하기 때문에

여러 방법을 동원하여 해결해보고자 한다.

 

먼저 이 글 (eagle 아님) 을 참고해서 진행해보고자 하였다.

TFLite 레포지토리를 클론하고, visualize.py를 bazel로 수행하라고 한다.

  bazel run //tensorflow/lite/tools:visualize \
       model.tflite \
       visualized_model.html

bazel로 수행하기 위해서, bazel을 설치하고자 하였다.

 

https://github.com/bazelbuild/bazelisk

bazelisk로 설치하는것을 권장하여, bazelisk로 설치하고자 했지만 이런, windows에서는 chocolately를 통해서 설치를 하라고 한다.

 

따라서 이 글을 참고하여 chocolately를 설치했다. Windows에서도 apt로 설치하는 느낌을 받게 해줬다.

 

Chocolately를 설치완료한 이후, bazelisk를 설치하였다.....이제보니 Release로 가서 exe를 받아도 되었을것같긴 하지만...

아무튼 Bazel 버전 체크도 확인하였고, 이제 첫번째 참조글을 이용하여 visualize를 해보도록 하겠다.

 

...

해당 코드가 동작하지 않아, 다른 방식을 찾아보았다.

https://github.com/eliberis/tflite-tools

 

GitHub - eliberis/tflite-tools: TFLite model analyzer & memory optimizer

TFLite model analyzer & memory optimizer. Contribute to eliberis/tflite-tools development by creating an account on GitHub.

github.com

꽤나 유용한 정보들을 제공해 주는것 같아 다음을 이용해보고자 하였다.

 

--------중단---------

tflite를 시각화 하는것은 의미가 없다고 생각되어서 중단했다.