IT/디버깅

한진정보통신 인턴 과제 - yolo8 모델을 android 모델로 변환 중 발생한 문제 - android 인식 안됨

happy_life 2024. 6. 5. 12:37

yolo8 모델의 학습을 끝내고, 인턴 프로젝트를 더 디벨롭하고 싶었습니다. 이에 카메라 앱에 적용해 실제로 모델을 실시간으로 탐지하도록 프로젝트를 진행하고자 하였습니다. 하지만 이 과정에서 발생한 문제들이 많았고 이를 정리하는 포스팅입니다. 

 

1. Android Studio에서 사용하기 위해 Tflite 모델로 적용 중 발생한 문제

 

Android studio에서 인식하지 못하는 문제

 

 

yolo8 모델의 가중치를 tflite모델로 변환하였으나, 안드로이드 스튜디오에서 인식하지 못하는 문제가 발생하였습니다.

 

문제 해결 과정

1. model 변환에 대한 의문 

yolo8 공식문서의 예시 segment 모델 가중치를 converting 해보았으나, 이 또한 android studio에서 인식하지 못하였습니다. 이에 최신 기술이기 때문에 변환을 아직 지원하지 않는 것이 아닐까 하는 생각이 들었습니다. 따라서, yolo8에 이슈를 올렸습니다.

질문/답변

 

 

이에 segment model같은 구조는 문제가 발생할 수 있다고 하였고, 추후 업데이트 예정이라는 답변을 얻었습니다.

 

ps. 밑에 댓글에 tflight 인자가 잘못되어서라고 댓글이 달렸는데, tflight로 인자를 잘못 넣으면 model 변환되기 전에 애초에 에러가 납니다... 제가 저 답변 이후에 이슈를 확인 안했는데 인자 잘못 넣은 문제로 결론나고 closed..되었더라고요 

 

2. model의 변환

이에 segment model을 detection model에 변환하여 문제를 해결하였습니다.