텐서플로를 활용한 EfficientDET 실습
지금까지는 EfficientDet의 이론적인 부분을 살펴보았습니다. 이를 바탕으로 EfficientDet을 구현하는 코드를 살펴볼 차례입니다. 특별히 이번엔 텐서플로 라이트(Tensorflow Lite)를 사용하여 EfficientDET 모델을 구현하는 방법을 살펴볼 것입니다. 텐서플로 라이트는 텐서플로의 경량화 버전으로, 모바일 및 임베디드 디바이스에서 머신 러닝 모델을 효율적으로 실행할 수 있도록 설계되었습니다.
먼저 다음처럼 필요한 라이브러리를 불러옵니다.
import tensorflow as tf
import tensorflow_hub as hub
import cv2
import numpy as np
import time
model = hub.load("https://tfhub.dev/tensorflow/efficientdet/lite0/detection/1") # ①
!wget https://github.com/Cobslab/imageBible/blob/main/image/test_image.png?raw=true -O test_image.png # ②