YOLOv8 to tflite

YOLOv8을 학습하여 tflite 파일로 모델을 추출하고 싶은데 에러가 계속 발생한다.

 

 

PyTorch: starting from 'runs\detect\train9\weights\best.pt' with input shape (1, 3, 640, 640) BCHW and output shape(s) (1, 84, 8400) (6.2 MB)
TensorFlow SavedModel: export failure ❌ 3.6s: Descriptors cannot be created directly.
If this call came from a _pb2.py file, your generated code is out of date and must be regenerated with protoc >= 3.19.0.
If you cannot immediately regenerate your protos, some other possible workarounds are:
 1. Downgrade the protobuf package to 3.20.x or lower.
 2. Set PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python (but this will use pure-Python parsing and will be much slower).

 

https://vhrehfdl.tistory.com/153

If this call came from a _pb2.py file, your generated code is out of date and must be regenerated with protoc >= 3.19.0.

 

여기 글을 참조해서 시도해보도록 하겠다.

 

pip install protobuf==3.20.*

이렇게 명령을 주고 다시 시도해보자.

 

 

또 다시 에러 몇가지를 더 설치해야하는데 왜자꾸 엑세스 거부가 되는지 모르겠네.

 

어쨋든 저기 나온 패키지를 설치하고 다시 수행해보자.

 

결국 pip install upgrade를 안하고 여기까지 온 결과 인듯하다.

 

하지만, 업그레이드 조차 제대로 되지 않아 다른 블로그를 참조했다.

 

https://domdom.tistory.com/562

[오류해결] ValueError: Unable to find resource t64.exe in package pip._vendor.distlib

 

어쨋든 pip upgrade와 모든 패키지 설치를 마치고 다시 yolo 모델을 tflite로 export를 해보겠다.

 

결과는 여전히 에러다.

 

뭐가 문제인지 유투브를 찾아보도록 하겠다.

 

https://www.youtube.com/watch?v=75LI9MI9eEo이고ㅓ

 

이거 보고 다시 해보도록 하겠다.

 

끝.