detect_objects 함수는 EfficientDET 모델의 탐지 결과를 시각적으로 확인할 수 있게 하며, 실제 애플리케이션에서 이미지 내 객체를 식별하고 위치를 파악하는 데 유용하게 사용될 수 있습니다. 또한 텐서플로 라이트 버전으로 실행이 되는 만큼, 어느 정도 짧은 시간 안에 탐지가 가능한지 확인해보겠습니다.

!wget https://github.com/Cobslab/imageBible/blob/main/image/test_image.png?raw=true -O test_image.png

img_array = cv2.cvtColor(cv2.imread('test_image.png'), cv2.COLOR_BGR2RGB) # ①
processed_img = detect_objects(model, img_array, 0.3, 100, True)          # ②

import matplotlib.pyplot as plt
plt.figure(figsize=(12, 12))
plt.imshow(processed_img) # ③

앞 코드는 test_image.png 파일을 불러와서 OpenCV를 사용하여 BGR에서 RGB로 색상 포맷을 변경한 후(①), detect_objects 함수를 호출하여 이미지 내 객체를 탐지하고(②), Matplotlib을 사용하여 처리된 이미지를 화면에 표시하는 과정으로 구성되어 있습니다(③). 출력 결과를 확인해보겠습니다.