object detection 3

[Tensorflow Lite] - 객체 탐지(Object Detection Model)만들기_3. 안드로이드에 Tensorflow Lite 모델 삽입

들어가며 지난 시간까지 데이터를 수집하고 Tensorflow Lite 모델을 만드는 과정까지 진행해보았다. 이번에는 이 모델 파일을 Android에 삽입해서 실제 구동까지 진행해보도록 하겠다. TFlite 모델에 메타데이터 추가하기 이 부분 때문에 필자는 몇 시간동안 삽질을 했다. 이전 장에서 만든 .tflite 파일을 그대로 안드로이드에 사용하게 되면 에러가 발생한다:(. 구글링을 하고 stack overflow에 업로드하는 등 여러 노력을 통해서 솔루션을 찾아냈다. 에러의 내용은 tflite모델에 NormalizationOptions이 추가되어 있어야 한다는 뜻이었다. 이는 메타 데이터의 일종인데 이를 삽입하기 위해선 추가적인 작업이 필요했다. 위 코드를 사용하면 필요한 메타 데이터를 추가 할 수 있..

Machine Learning 2021.03.17

[Tensorflow Lite] - 객체 탐지(Object Detection Model)만들기_2. 모델 학습

들어가며 지난 시간에는 roboflow를 사용해서 데이터를 업로드하여 처리하는 방법에 대해서 알아보았다. 이번에는 데이터와 Tensorflow 프레임워크를 사용해서 모델을 만들어보도록 하겠다. 추가적으로 컨버터를 사용해서 Tensorflow Lite 모델까지 변환하는 방법에 대해서 알아보도록 하겠다. (이 장에서는 필수적으로 수정해야 할 부분에 대해서만 언급하겠다. 코드에 대해서 자세히 알고 싶다면 Colab내부의 주석을 참고하도록 하자) 데이터 다운로드 colab.research.google.com/drive/1aIobwtqWggRbLbe606_WzVGtFiWZT9tr?usp=sharing Roboflow-TFLite-Object-Detection_Bicycle_Helmet.ipynb Colaborat..

Machine Learning 2021.03.17

[Tensorflow Lite] - 객체 탐지(Object Detection Model)만들기_1. 데이터 처리

들어가며 최근에 어떤 프로젝트를 진행하게 되면서 안전모를 착용했는지 확인하기 위한 모바일용 머신러닝 모델이 필요하게 되었다. 3일이라는 짧은 시간 안에 프로토타입을 만들게 되었는데, 진행하면서 알게 된 에러 사항에 대해서 공유하기 위해서 이 글을 작성하니 모바일용 객체 탐지 모델을 만드는 사람들이 참고해서 쉽게 사용했으면 좋겠다. 혹시 결과물이 바로 보고 싶은 사람은 아래의 링크를 참고하란다. github.com/BEOKS/Bicycle-Helmet-Wearing-Detection BEOKS/Bicycle-Helmet-Wearing-Detection Tensorflow lite model that detects bicycle helmet wearing and Android demo application ..

Machine Learning 2021.03.17